├── .gitignore ├── LICENSE ├── README.md ├── configs └── freesolo │ ├── freemask.yaml │ └── freesolo_30k.yaml ├── dataseed ├── COCO_supervision.txt └── generate_seeds.py ├── demo ├── demo.py ├── inference_freemask.py ├── merge_json.py ├── pipeline.jpg ├── predictor.py └── vis.jpg ├── freesolo ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── build.py │ ├── common.py │ ├── dataset_mapper.py │ ├── datasets │ │ ├── builtin.py │ │ └── coco.py │ ├── detection_utils.py │ └── transforms │ │ └── augmentation_impl.py ├── engine │ └── trainer.py └── modeling │ └── solov2 │ ├── __init__.py │ ├── loss.py │ ├── pseudo_solov2.py │ ├── solov2.py │ └── utils.py ├── gen_pseudo_labels.sh ├── inference_freemask.sh ├── test.sh ├── tools ├── convert-pretrain-to-detectron2.py ├── eval_cocoapi.py ├── gen_pseudo_labels.py ├── merge_train2017_unlabeled2017.py ├── split_coco_train_sup10.py ├── split_coco_train_sup10_usemask.py ├── visualize.sh ├── visualize_data.py └── visualize_json_results.py ├── train.sh ├── train_net.py └── train_pl.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/README.md -------------------------------------------------------------------------------- /configs/freesolo/freemask.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/configs/freesolo/freemask.yaml -------------------------------------------------------------------------------- /configs/freesolo/freesolo_30k.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/configs/freesolo/freesolo_30k.yaml -------------------------------------------------------------------------------- /dataseed/COCO_supervision.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/dataseed/COCO_supervision.txt -------------------------------------------------------------------------------- /dataseed/generate_seeds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/dataseed/generate_seeds.py -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/inference_freemask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/inference_freemask.py -------------------------------------------------------------------------------- /demo/merge_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/merge_json.py -------------------------------------------------------------------------------- /demo/pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/pipeline.jpg -------------------------------------------------------------------------------- /demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/predictor.py -------------------------------------------------------------------------------- /demo/vis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/demo/vis.jpg -------------------------------------------------------------------------------- /freesolo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/__init__.py -------------------------------------------------------------------------------- /freesolo/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/config.py -------------------------------------------------------------------------------- /freesolo/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/__init__.py -------------------------------------------------------------------------------- /freesolo/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/build.py -------------------------------------------------------------------------------- /freesolo/data/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/common.py -------------------------------------------------------------------------------- /freesolo/data/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/dataset_mapper.py -------------------------------------------------------------------------------- /freesolo/data/datasets/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/datasets/builtin.py -------------------------------------------------------------------------------- /freesolo/data/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/datasets/coco.py -------------------------------------------------------------------------------- /freesolo/data/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/detection_utils.py -------------------------------------------------------------------------------- /freesolo/data/transforms/augmentation_impl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/data/transforms/augmentation_impl.py -------------------------------------------------------------------------------- /freesolo/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/engine/trainer.py -------------------------------------------------------------------------------- /freesolo/modeling/solov2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/modeling/solov2/__init__.py -------------------------------------------------------------------------------- /freesolo/modeling/solov2/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/modeling/solov2/loss.py -------------------------------------------------------------------------------- /freesolo/modeling/solov2/pseudo_solov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/modeling/solov2/pseudo_solov2.py -------------------------------------------------------------------------------- /freesolo/modeling/solov2/solov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/modeling/solov2/solov2.py -------------------------------------------------------------------------------- /freesolo/modeling/solov2/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/freesolo/modeling/solov2/utils.py -------------------------------------------------------------------------------- /gen_pseudo_labels.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/gen_pseudo_labels.sh -------------------------------------------------------------------------------- /inference_freemask.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/inference_freemask.sh -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/test.sh -------------------------------------------------------------------------------- /tools/convert-pretrain-to-detectron2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/convert-pretrain-to-detectron2.py -------------------------------------------------------------------------------- /tools/eval_cocoapi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/eval_cocoapi.py -------------------------------------------------------------------------------- /tools/gen_pseudo_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/gen_pseudo_labels.py -------------------------------------------------------------------------------- /tools/merge_train2017_unlabeled2017.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/merge_train2017_unlabeled2017.py -------------------------------------------------------------------------------- /tools/split_coco_train_sup10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/split_coco_train_sup10.py -------------------------------------------------------------------------------- /tools/split_coco_train_sup10_usemask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/split_coco_train_sup10_usemask.py -------------------------------------------------------------------------------- /tools/visualize.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/visualize.sh -------------------------------------------------------------------------------- /tools/visualize_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/visualize_data.py -------------------------------------------------------------------------------- /tools/visualize_json_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/tools/visualize_json_results.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/train.sh -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/train_net.py -------------------------------------------------------------------------------- /train_pl.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVlabs/FreeSOLO/HEAD/train_pl.sh --------------------------------------------------------------------------------