fovi.models package

Optional model architectures and inference loading (fovi[models]).

Modules

• fovi.models.alexnet
  • ConvBlock
    • ConvBlock.__init__()
  • get_backbone()
  • get_repr_size()
  • build_model()
  • alexnet2023_baseline()
• fovi.models.architectures
  • alexnet2023()
  • fovi_alexnet2023()
  • resnet18()
  • resnet34()
  • resnet50()
  • matched_resnet18()
  • matched_resnet50()
  • fovi_resnet()
  • fovi_resnet9()
  • fovi_resnet9_lowres()
  • fovi_resnet18()
  • fovi_resnet50()
  • fovi_resnet18_lora()
  • fovi_resnet18_lowres()
  • fovi_resnet9_dwsep()
  • fovi_resnet9_dwsep_lowres()
  • fovi_resnet18_dwsep()
  • fovi_resnet18_dwsep_lowres()
  • fovi_vit()
  • fovi_vit_base()
  • fovi_vit_small()
  • fovi_vit_tiny()
  • vit()
  • vit_base()
  • vit_small()
  • vit_tiny()
  • vit_custom()
  • fovi_vit_custom()
  • fovi_dinov3()
  • arch_wrapper()
  • ArchitectureRegistry
    • ArchitectureRegistry.__init__()
    • ArchitectureRegistry.register()
    • ArchitectureRegistry.get()
    • ArchitectureRegistry.has()
    • ArchitectureRegistry.list_architectures()
• fovi.models.convnextv2
  • LayerNorm
    • LayerNorm.__init__()
    • LayerNorm.forward()
  • GRN
    • GRN.__init__()
    • GRN.forward()
  • Block
    • Block.__init__()
    • Block.forward()
  • ConvNeXtV2
    • ConvNeXtV2.__init__()
    • ConvNeXtV2.forward_features()
    • ConvNeXtV2.forward()
  • convnextv2_atto()
  • convnextv2_femto()
  • convnext_pico()
  • convnextv2_nano()
  • convnextv2_tiny()
  • convnextv2_base()
  • convnextv2_large()
  • convnextv2_huge()
• fovi.models.dinov3
  • load_dinov3()
  • build_fovi_dinov3()
  • prep_fovi_dinov3_finetuning()
• fovi.models.fovinet
  • FoviNet
    • FoviNet.__init__()
    • FoviNet.init_fixation_system()
    • FoviNet.get_in_channels()
    • FoviNet.get_repr_sizes()
    • FoviNet.to()
    • FoviNet.forward_ssl()
    • FoviNet.forward_supervised()
    • FoviNet.forward()
    • FoviNet.get_transforms()
    • FoviNet.setup_activation_hooks()
    • FoviNet.get_captured_activations()
    • FoviNet.cleanup_activation_hooks()
    • FoviNet.list_available_layers()
    • FoviNet.get_activations()
    • FoviNet.list_knn_layers()
    • FoviNet.get_num_coords()
• fovi.models.hub
  • download_model()
• fovi.models.knnalexnet
  • KNNAlexNetBlock
    • KNNAlexNetBlock.__init__()
    • KNNAlexNetBlock.forward()
  • KNNAlexNet
    • KNNAlexNet.__init__()
    • KNNAlexNet.forward()
• fovi.models.knnconvnextv2
  • GRN
    • GRN.__init__()
    • GRN.forward()
  • LayerNorm
    • LayerNorm.__init__()
    • LayerNorm.forward()
  • Block
    • Block.__init__()
    • Block.forward()
  • ConvNeXtV2
    • ConvNeXtV2.__init__()
    • ConvNeXtV2.forward_features()
    • ConvNeXtV2.forward()
  • knnconvnextv2_atto()
  • knnconvnextv2_femto()
  • knnconvnextv2_pico()
  • knnconvnextv2_nano()
  • knnconvnextv2_tiny()
  • knnconvnextv2_base()
  • knnconvnextv2_large()
  • knnconvnextv2_huge()
• fovi.models.knnresnet
  • KNNResNetBasicBlock
    • KNNResNetBasicBlock.__init__()
    • KNNResNetBasicBlock.expansion
    • KNNResNetBasicBlock.forward()
  • KNNResNetBottleneck
    • KNNResNetBottleneck.expansion
    • KNNResNetBottleneck.__init__()
    • KNNResNetBottleneck.forward()
  • KNNResNet
    • KNNResNet.__init__()
    • KNNResNet._make_layer()
    • KNNResNet.forward()
• fovi.models.knnvit
  • KNNPatchEmbedding
    • KNNPatchEmbedding.__init__()
    • KNNPatchEmbedding.forward()
  • PartitioningPatchEmbedding
    • PartitioningPatchEmbedding.__init__()
  • KNNPartitioningPatchEmbedding
    • KNNPartitioningPatchEmbedding._stable_neighbor_order()
    • KNNPartitioningPatchEmbedding.__init__()
  • KNNViT
    • KNNViT.__init__()
    • KNNViT.forward()
  • FoviDinoV3RoPE
    • FoviDinoV3RoPE.inv_freq
    • FoviDinoV3RoPE.__init__()
    • FoviDinoV3RoPE.forward()
  • resample_patch_embed_conv()
• fovi.models.loading
  • Sequence
    • Sequence.count()
    • Sequence.index()
  • Path
    • Path.cwd()
    • Path.home()
    • Path.samefile()
    • Path.iterdir()
    • Path.glob()
    • Path.rglob()
    • Path.absolute()
    • Path.resolve()
    • Path.stat()
    • Path.owner()
    • Path.group()
    • Path.open()
    • Path.read_bytes()
    • Path.read_text()
    • Path.write_bytes()
    • Path.write_text()
    • Path.readlink()
    • Path.touch()
    • Path.mkdir()
    • Path.chmod()
    • Path.lchmod()
    • Path.unlink()
    • Path.rmdir()
    • Path.lstat()
    • Path.rename()
    • Path.replace()
    • Path.symlink_to()
    • Path.hardlink_to()
    • Path.link_to()
    • Path.exists()
    • Path.is_dir()
    • Path.is_file()
    • Path.is_mount()
    • Path.is_symlink()
    • Path.is_block_device()
    • Path.is_char_device()
    • Path.is_fifo()
    • Path.is_socket()
    • Path.expanduser()
  • GlobalHydra
    • GlobalHydra.__init__()
    • GlobalHydra.initialize()
    • GlobalHydra.config_loader()
    • GlobalHydra.is_initialized()
    • GlobalHydra.clear()
    • GlobalHydra.instance()
    • GlobalHydra.set_instance()
  • DictConfig
    • DictConfig.__init__()
    • DictConfig.copy()
    • DictConfig.get()
    • DictConfig.pop()
    • DictConfig.keys()
    • DictConfig.items()
    • DictConfig.setdefault()
    • DictConfig.items_ex()
    • DictConfig._promote()
    • DictConfig._to_object()
  • OmegaConf
    • OmegaConf.__init__()
    • OmegaConf.structured()
    • OmegaConf.create()
    • OmegaConf.load()
    • OmegaConf.save()
    • OmegaConf.from_cli()
    • OmegaConf.from_dotlist()
    • OmegaConf.merge()
    • OmegaConf.unsafe_merge()
    • OmegaConf.register_resolver()
    • OmegaConf.legacy_register_resolver()
    • OmegaConf.register_new_resolver()
    • OmegaConf.has_resolver()
    • OmegaConf.clear_resolvers()
    • OmegaConf.clear_resolver()
    • OmegaConf.get_cache()
    • OmegaConf.set_cache()
    • OmegaConf.clear_cache()
    • OmegaConf.copy_cache()
    • OmegaConf.set_readonly()
    • OmegaConf.is_readonly()
    • OmegaConf.set_struct()
    • OmegaConf.is_struct()
    • OmegaConf.masked_copy()
    • OmegaConf.to_container()
    • OmegaConf.to_object()
    • OmegaConf.is_missing()
    • OmegaConf.is_interpolation()
    • OmegaConf.is_list()
    • OmegaConf.is_dict()
    • OmegaConf.is_config()
    • OmegaConf.get_type()
    • OmegaConf.select()
    • OmegaConf.update()
    • OmegaConf.to_yaml()
    • OmegaConf.resolve()
    • OmegaConf.missing_keys()
  • open_dict()
  • HiddenPrints
    • HiddenPrints.__init__()
  • default_model_dirs()
  • load_sharded_state_dict()
  • load_config()
  • find_config()
  • get_model_from_base_fn()
• fovi.models.pretrained_resnet
  • Iterable
  • ResNet18_Weights
    • ResNet18_Weights.IMAGENET1K_V1
    • ResNet18_Weights.DEFAULT
  • resnet18()
    • ResNet18_Weights
      • ResNet18_Weights.IMAGENET1K_V1
      • ResNet18_Weights.DEFAULT
  • apply_lora()
  • KNNConvLayer
    • KNNConvLayer.__init__()
    • KNNConvLayer._init_conv_like()
    • KNNConvLayer._pad_and_gather_knns()
    • KNNConvLayer._apply_local_rf()
    • KNNConvLayer._apply_local_rf_to_weights()
    • KNNConvLayer.forward()
    • KNNConvLayer.clear_optimized_cache()
    • KNNConvLayer.compute_reference_coords()
    • KNNConvLayer.compute_local_rf()
    • KNNConvLayer.load_conv2d_weights()
    • KNNConvLayer.load_conv3d_weights()
  • KNNResNet
    • KNNResNet.__init__()
    • KNNResNet._make_layer()
    • KNNResNet.forward()
  • flattened_basic_blocks()
  • load_torchvision_resnet18()
  • load_torchvision_resnet_backbone()
  • prep_fovi_resnet_finetuning()
• fovi.models.probes
  • LinearProbe
    • LinearProbe.probe
    • LinearProbe.dropout
    • LinearProbe.__init__()
    • LinearProbe.forward()
  • FoviNetProbe
    • FoviNetProbe.fix_agg
    • FoviNetProbe.num_features
    • FoviNetProbe.num_classes
    • FoviNetProbe.dropout
    • FoviNetProbe.__init__()
    • FoviNetProbe._init_fix_agg()
    • FoviNetProbe.forward()
  • FoviNetProbes
    • FoviNetProbes.mlp_spec
    • FoviNetProbes.fix_agg
    • FoviNetProbes.num_classes
    • FoviNetProbes.dropout
    • FoviNetProbes.probes
    • FoviNetProbes.__init__()
    • FoviNetProbes._init_probes()
    • FoviNetProbes.forward()
• fovi.models.resnet
  • conv3x3_polar()
  • conv_polar()
  • BasicBlockPolar
    • BasicBlockPolar.expansion
    • BasicBlockPolar.expansion
    • BasicBlockPolar.__init__()
    • BasicBlockPolar.forward()
  • BottleneckPolar
    • BottleneckPolar.expansion
    • BottleneckPolar.expansion
    • BottleneckPolar.__init__()
    • BottleneckPolar.forward()
  • ResNet
    • ResNet.__init__()
  • resnet18()
  • resnet34()
  • resnet50()
  • resnet101()
  • get_repr_size()
  • resnet_ssl()
• fovi.models.vit
  • apply_2d_rotary_pos_emb()
  • PositionalEncoding
    • PositionalEncoding.__init__()
    • PositionalEncoding.forward()
  • RoPEPositionalEncoding
    • RoPEPositionalEncoding.__init__()
    • RoPEPositionalEncoding.forward()
  • PatchEmbedding
    • PatchEmbedding.__init__()
    • PatchEmbedding.forward()
  • MultiHeadSelfAttention
    • MultiHeadSelfAttention.__init__()
    • MultiHeadSelfAttention.forward()
  • RoPEMultiHeadSelfAttention
    • RoPEMultiHeadSelfAttention.__init__()
    • RoPEMultiHeadSelfAttention.forward()
  • TransformerBlock
    • TransformerBlock.__init__()
    • TransformerBlock.forward()
  • RoPETransformerBlock
    • RoPETransformerBlock.__init__()
  • VisionTransformer
    • VisionTransformer.__init__()
    • VisionTransformer._init_weights()
    • VisionTransformer.forward()