├── .gitignore ├── README.md ├── SECURITY.md ├── dataset ├── __init__.py ├── annotation_centroids.npy ├── davis.py ├── split_trainval.py └── transforms.py ├── figure └── fig1.png ├── inference.py ├── lib ├── __init__.py ├── loss.py ├── predict.py └── utils.py ├── main.py ├── modeling ├── __init__.py ├── backbone │ ├── __init__.py │ └── resnet.py ├── network.py └── sync_batchnorm │ ├── __init__.py │ ├── batchnorm.py │ ├── comm.py │ ├── replicate.py │ └── unittest.py ├── test_epochs.py └── test_epochs.sh /.gitignore: -------------------------------------------------------------------------------- 1 | *.npy 2 | __pycache__ 3 | .vscode 4 | .idea 5 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /SECURITY.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/SECURITY.md -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/annotation_centroids.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/dataset/annotation_centroids.npy -------------------------------------------------------------------------------- /dataset/davis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/dataset/davis.py -------------------------------------------------------------------------------- /dataset/split_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/dataset/split_trainval.py -------------------------------------------------------------------------------- /dataset/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/dataset/transforms.py -------------------------------------------------------------------------------- /figure/fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/figure/fig1.png -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/inference.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/lib/loss.py -------------------------------------------------------------------------------- /lib/predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/lib/predict.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/lib/utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/main.py -------------------------------------------------------------------------------- /modeling/__init__.py: -------------------------------------------------------------------------------- 1 | from .network import VOSNet 2 | -------------------------------------------------------------------------------- /modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modeling/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/backbone/resnet.py -------------------------------------------------------------------------------- /modeling/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/network.py -------------------------------------------------------------------------------- /modeling/sync_batchnorm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/sync_batchnorm/__init__.py -------------------------------------------------------------------------------- /modeling/sync_batchnorm/batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/sync_batchnorm/batchnorm.py -------------------------------------------------------------------------------- /modeling/sync_batchnorm/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/sync_batchnorm/comm.py -------------------------------------------------------------------------------- /modeling/sync_batchnorm/replicate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/sync_batchnorm/replicate.py -------------------------------------------------------------------------------- /modeling/sync_batchnorm/unittest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/modeling/sync_batchnorm/unittest.py -------------------------------------------------------------------------------- /test_epochs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/microsoft/transductive-vos.pytorch/HEAD/test_epochs.py -------------------------------------------------------------------------------- /test_epochs.sh: -------------------------------------------------------------------------------- 1 | mpiexec -n 16 python test_epochs.py 2 | --------------------------------------------------------------------------------