├── .gitignore ├── CHANGELOG.md ├── LICENSE ├── README.md ├── backbone.py ├── data ├── __init__.py ├── coco.py ├── config.py ├── grid.npy ├── scripts │ ├── COCO.sh │ ├── COCO_test.sh │ └── mix_sets.py ├── yolact_example_0.png ├── yolact_example_1.png └── yolact_example_2.png ├── environment.yml ├── eval.py ├── external └── DCNv2 │ ├── LICENSE │ ├── README.md │ ├── __init__.py │ ├── dcn_v2.py │ ├── setup.py │ ├── src │ ├── cpu │ │ ├── dcn_v2_cpu.cpp │ │ └── vision.h │ ├── cuda │ │ ├── dcn_v2_cuda.cu │ │ ├── dcn_v2_im2col_cuda.cu │ │ ├── dcn_v2_im2col_cuda.h │ │ ├── dcn_v2_psroi_pooling_cuda.cu │ │ └── vision.h │ ├── dcn_v2.h │ └── vision.cpp │ └── test.py ├── layers ├── __init__.py ├── box_utils.py ├── functions │ ├── __init__.py │ └── detection.py ├── interpolate.py ├── modules │ ├── __init__.py │ └── multibox_loss.py └── output_utils.py ├── run_coco_eval.py ├── scripts ├── augment_bbox.py ├── bbox_recall.py ├── cluster_bbox_sizes.py ├── compute_masks.py ├── convert_darknet.py ├── convert_sbd.py ├── eval.sh ├── make_grid.py ├── optimize_bboxes.py ├── parse_eval.py ├── plot_loss.py ├── resume.sh ├── save_bboxes.py ├── train.sh └── unpack_statedict.py ├── train.py ├── utils ├── __init__.py ├── augmentations.py ├── cython_nms.pyx ├── functions.py ├── logger.py ├── nvinfo.py └── timer.py ├── web ├── css │ ├── index.css │ ├── list.css │ ├── toggle.css │ └── viewer.css ├── dets │ ├── ssd300.json │ ├── ssd550.json │ ├── ssd550_resnet101.json │ ├── test.json │ ├── yolact_base.json │ ├── yolact_darknet53.json │ ├── yolact_im700.json │ ├── yolact_resnet101_conv4.json │ ├── yolact_resnet101_maskrcnn.json │ ├── yolact_resnet101_maskrcnn_1.json │ ├── yolact_resnet50.json │ ├── yrm12.json │ ├── yrm13.json │ ├── yrm16_2.json │ ├── yrm18.json │ ├── yrm19.json │ ├── yrm21.json │ ├── yrm25_b.json │ ├── yrm28_2_perfect.json │ ├── yrm35_crop.json │ └── yrm35_retina.json ├── index.html ├── iou.html ├── scripts │ ├── index.js │ ├── iou.js │ ├── jquery.js │ ├── js.cookie.js │ ├── utils.js │ └── viewer.js ├── server.py └── viewer.html └── yolact.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/README.md -------------------------------------------------------------------------------- /backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/backbone.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/coco.py -------------------------------------------------------------------------------- /data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/config.py -------------------------------------------------------------------------------- /data/grid.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/grid.npy -------------------------------------------------------------------------------- /data/scripts/COCO.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/scripts/COCO.sh -------------------------------------------------------------------------------- /data/scripts/COCO_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/scripts/COCO_test.sh -------------------------------------------------------------------------------- /data/scripts/mix_sets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/scripts/mix_sets.py -------------------------------------------------------------------------------- /data/yolact_example_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/yolact_example_0.png -------------------------------------------------------------------------------- /data/yolact_example_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/yolact_example_1.png -------------------------------------------------------------------------------- /data/yolact_example_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/data/yolact_example_2.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/eval.py -------------------------------------------------------------------------------- /external/DCNv2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/LICENSE -------------------------------------------------------------------------------- /external/DCNv2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/README.md -------------------------------------------------------------------------------- /external/DCNv2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /external/DCNv2/dcn_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/dcn_v2.py -------------------------------------------------------------------------------- /external/DCNv2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/setup.py -------------------------------------------------------------------------------- /external/DCNv2/src/cpu/dcn_v2_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cpu/dcn_v2_cpu.cpp -------------------------------------------------------------------------------- /external/DCNv2/src/cpu/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cpu/vision.h -------------------------------------------------------------------------------- /external/DCNv2/src/cuda/dcn_v2_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cuda/dcn_v2_cuda.cu -------------------------------------------------------------------------------- /external/DCNv2/src/cuda/dcn_v2_im2col_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cuda/dcn_v2_im2col_cuda.cu -------------------------------------------------------------------------------- /external/DCNv2/src/cuda/dcn_v2_im2col_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cuda/dcn_v2_im2col_cuda.h -------------------------------------------------------------------------------- /external/DCNv2/src/cuda/dcn_v2_psroi_pooling_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cuda/dcn_v2_psroi_pooling_cuda.cu -------------------------------------------------------------------------------- /external/DCNv2/src/cuda/vision.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/cuda/vision.h -------------------------------------------------------------------------------- /external/DCNv2/src/dcn_v2.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/dcn_v2.h -------------------------------------------------------------------------------- /external/DCNv2/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/src/vision.cpp -------------------------------------------------------------------------------- /external/DCNv2/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/external/DCNv2/test.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/__init__.py -------------------------------------------------------------------------------- /layers/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/box_utils.py -------------------------------------------------------------------------------- /layers/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/functions/__init__.py -------------------------------------------------------------------------------- /layers/functions/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/functions/detection.py -------------------------------------------------------------------------------- /layers/interpolate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/interpolate.py -------------------------------------------------------------------------------- /layers/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/modules/__init__.py -------------------------------------------------------------------------------- /layers/modules/multibox_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/modules/multibox_loss.py -------------------------------------------------------------------------------- /layers/output_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/layers/output_utils.py -------------------------------------------------------------------------------- /run_coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/run_coco_eval.py -------------------------------------------------------------------------------- /scripts/augment_bbox.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/augment_bbox.py -------------------------------------------------------------------------------- /scripts/bbox_recall.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/bbox_recall.py -------------------------------------------------------------------------------- /scripts/cluster_bbox_sizes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/cluster_bbox_sizes.py -------------------------------------------------------------------------------- /scripts/compute_masks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/compute_masks.py -------------------------------------------------------------------------------- /scripts/convert_darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/convert_darknet.py -------------------------------------------------------------------------------- /scripts/convert_sbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/convert_sbd.py -------------------------------------------------------------------------------- /scripts/eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/eval.sh -------------------------------------------------------------------------------- /scripts/make_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/make_grid.py -------------------------------------------------------------------------------- /scripts/optimize_bboxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/optimize_bboxes.py -------------------------------------------------------------------------------- /scripts/parse_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/parse_eval.py -------------------------------------------------------------------------------- /scripts/plot_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/plot_loss.py -------------------------------------------------------------------------------- /scripts/resume.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/resume.sh -------------------------------------------------------------------------------- /scripts/save_bboxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/save_bboxes.py -------------------------------------------------------------------------------- /scripts/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/train.sh -------------------------------------------------------------------------------- /scripts/unpack_statedict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/scripts/unpack_statedict.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/cython_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/cython_nms.pyx -------------------------------------------------------------------------------- /utils/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/functions.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/nvinfo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/nvinfo.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/utils/timer.py -------------------------------------------------------------------------------- /web/css/index.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/css/index.css -------------------------------------------------------------------------------- /web/css/list.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/css/list.css -------------------------------------------------------------------------------- /web/css/toggle.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/css/toggle.css -------------------------------------------------------------------------------- /web/css/viewer.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/css/viewer.css -------------------------------------------------------------------------------- /web/dets/ssd300.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/ssd300.json -------------------------------------------------------------------------------- /web/dets/ssd550.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/ssd550.json -------------------------------------------------------------------------------- /web/dets/ssd550_resnet101.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/ssd550_resnet101.json -------------------------------------------------------------------------------- /web/dets/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/test.json -------------------------------------------------------------------------------- /web/dets/yolact_base.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_base.json -------------------------------------------------------------------------------- /web/dets/yolact_darknet53.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_darknet53.json -------------------------------------------------------------------------------- /web/dets/yolact_im700.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_im700.json -------------------------------------------------------------------------------- /web/dets/yolact_resnet101_conv4.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_resnet101_conv4.json -------------------------------------------------------------------------------- /web/dets/yolact_resnet101_maskrcnn.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_resnet101_maskrcnn.json -------------------------------------------------------------------------------- /web/dets/yolact_resnet101_maskrcnn_1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_resnet101_maskrcnn_1.json -------------------------------------------------------------------------------- /web/dets/yolact_resnet50.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yolact_resnet50.json -------------------------------------------------------------------------------- /web/dets/yrm12.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm12.json -------------------------------------------------------------------------------- /web/dets/yrm13.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm13.json -------------------------------------------------------------------------------- /web/dets/yrm16_2.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm16_2.json -------------------------------------------------------------------------------- /web/dets/yrm18.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm18.json -------------------------------------------------------------------------------- /web/dets/yrm19.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm19.json -------------------------------------------------------------------------------- /web/dets/yrm21.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm21.json -------------------------------------------------------------------------------- /web/dets/yrm25_b.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm25_b.json -------------------------------------------------------------------------------- /web/dets/yrm28_2_perfect.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm28_2_perfect.json -------------------------------------------------------------------------------- /web/dets/yrm35_crop.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm35_crop.json -------------------------------------------------------------------------------- /web/dets/yrm35_retina.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/dets/yrm35_retina.json -------------------------------------------------------------------------------- /web/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/index.html -------------------------------------------------------------------------------- /web/iou.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/iou.html -------------------------------------------------------------------------------- /web/scripts/index.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/index.js -------------------------------------------------------------------------------- /web/scripts/iou.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/iou.js -------------------------------------------------------------------------------- /web/scripts/jquery.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/jquery.js -------------------------------------------------------------------------------- /web/scripts/js.cookie.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/js.cookie.js -------------------------------------------------------------------------------- /web/scripts/utils.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/utils.js -------------------------------------------------------------------------------- /web/scripts/viewer.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/scripts/viewer.js -------------------------------------------------------------------------------- /web/server.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/server.py -------------------------------------------------------------------------------- /web/viewer.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/web/viewer.html -------------------------------------------------------------------------------- /yolact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dbolya/yolact/HEAD/yolact.py --------------------------------------------------------------------------------