MCPcopy Index your code
hub / github.com/HobbitLong/PyContrast

github.com/HobbitLong/PyContrast @main sqlite

repository ↗ · DeepWiki ↗
244 symbols 551 edges 23 files 89 documented · 36%
README

PyContrast

This repo lists recent contrastive learning papers, and includes code for many of them.

Paper List

Find awesome contrastive learning paper.

PyTorch Code

For the reference implementation of SoTA methods, (e.g., InstDis, CMC, MoCo, ...), see pycontrast.

Pre-trained Models

A set of ImageNet unsupervised pre-training models are provided. Find them in model zoo.

Object Detection

Unsupervised pre-training models outperform supervised pre-training models on PASCAL VOC and COCO. Find in detection.

Core symbols most depended-on inside this repo

cuda
called by 26
pycontrast/memory/alias_multinomial.py
update
called by 21
pycontrast/learning/util.py
_resnet
called by 14
pycontrast/networks/resnet.py
_make_layer
called by 8
pycontrast/networks/resnest.py
contrast
called by 8
pycontrast/datasets/RandAugment.py
_compute_loss_accuracy
called by 8
pycontrast/learning/contrast_trainer.py
_check_args_tf
called by 7
pycontrast/datasets/RandAugment.py
_update_memory
called by 6
pycontrast/memory/mem_bank.py

Shape

Method 123
Function 74
Class 47

Languages

Python100%

Modules by API surface

pycontrast/datasets/RandAugment.py42 symbols
pycontrast/datasets/util.py32 symbols
pycontrast/networks/resnet.py27 symbols
pycontrast/networks/resnet_cmc.py19 symbols
pycontrast/networks/resnest.py19 symbols
pycontrast/learning/contrast_trainer.py16 symbols
pycontrast/networks/build_backbone.py15 symbols
pycontrast/memory/mem_moco.py11 symbols
pycontrast/memory/mem_bank.py10 symbols
pycontrast/learning/linear_trainer.py9 symbols
pycontrast/options/base_options.py7 symbols
pycontrast/networks/util.py7 symbols

Dependencies from manifests, versioned

numpy1.14.3 · 1×
pillow9.3.0 · 1×
scikit-image0.14.3 · 1×
scipy1.2.1 · 1×
torchvision0.2.1 · 1×

For agents

$ claude mcp add PyContrast \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact