├── README.md ├── code ├── datasets │ ├── VOCdevkit-matlab-wrapper │ │ ├── get_voc_opts.m │ │ ├── voc_eval.m │ │ └── xVOCap.m │ ├── __init__.py │ ├── corloc.py │ ├── pascal_voc.py │ └── voc_dataset_evaluator.py ├── layers │ ├── __init__.py │ ├── losses │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ ├── center_loss.cpython-36.pyc │ │ │ ├── mil_loss.cpython-36.pyc │ │ │ └── oicr_losses.cpython-36.pyc │ │ ├── mil_loss.py │ │ └── oicr_losses.py │ ├── mil.py │ ├── refinement │ │ ├── __init__.py │ │ ├── __pycache__ │ │ │ ├── __init__.cpython-36.pyc │ │ │ └── oicr.cpython-36.pyc │ │ └── oicr.py │ ├── refinement_agents.py │ └── roi_pooling │ │ ├── __init__.py │ │ ├── __pycache__ │ │ ├── __init__.cpython-36.pyc │ │ ├── fc_relu.cpython-36.pyc │ │ ├── roi_pool.cpython-36.pyc │ │ └── roi_pooling.cpython-36.pyc │ │ └── roi_pool.py ├── models │ ├── __init__.py │ ├── midn_oicr.py │ └── vgg16.py ├── tasks │ ├── __init__.py │ ├── config.py │ ├── init_paths.py │ ├── test.py │ ├── train.py │ └── visualize.py └── utils │ ├── __init__.py │ ├── boxes.py │ ├── data_samplers.py │ ├── data_transforms.py │ ├── detectron_weight_helper.py │ ├── logging.py │ ├── misc.py │ ├── optimizer_utils.py │ ├── test_utils.py │ ├── timer.py │ ├── training_stats.py │ └── vgg_weights_helper.py ├── configs └── baselines │ ├── vgg16_voc2007.yaml │ └── vgg16_voc2012.yaml └── requirements.txt /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /code/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m -------------------------------------------------------------------------------- /code/datasets/VOCdevkit-matlab-wrapper/voc_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/VOCdevkit-matlab-wrapper/voc_eval.m -------------------------------------------------------------------------------- /code/datasets/VOCdevkit-matlab-wrapper/xVOCap.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/VOCdevkit-matlab-wrapper/xVOCap.m -------------------------------------------------------------------------------- /code/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/datasets/corloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/corloc.py -------------------------------------------------------------------------------- /code/datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/pascal_voc.py -------------------------------------------------------------------------------- /code/datasets/voc_dataset_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/datasets/voc_dataset_evaluator.py -------------------------------------------------------------------------------- /code/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/layers/losses/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/layers/losses/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/losses/__pycache__/center_loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/__pycache__/center_loss.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/losses/__pycache__/mil_loss.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/__pycache__/mil_loss.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/losses/__pycache__/oicr_losses.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/__pycache__/oicr_losses.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/losses/mil_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/mil_loss.py -------------------------------------------------------------------------------- /code/layers/losses/oicr_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/losses/oicr_losses.py -------------------------------------------------------------------------------- /code/layers/mil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/mil.py -------------------------------------------------------------------------------- /code/layers/refinement/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/layers/refinement/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/refinement/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/refinement/__pycache__/oicr.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/refinement/__pycache__/oicr.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/refinement/oicr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/refinement/oicr.py -------------------------------------------------------------------------------- /code/layers/refinement_agents.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/refinement_agents.py -------------------------------------------------------------------------------- /code/layers/roi_pooling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/layers/roi_pooling/__pycache__/__init__.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/roi_pooling/__pycache__/__init__.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/roi_pooling/__pycache__/fc_relu.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/roi_pooling/__pycache__/fc_relu.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/roi_pooling/__pycache__/roi_pool.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/roi_pooling/__pycache__/roi_pool.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/roi_pooling/__pycache__/roi_pooling.cpython-36.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/roi_pooling/__pycache__/roi_pooling.cpython-36.pyc -------------------------------------------------------------------------------- /code/layers/roi_pooling/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/layers/roi_pooling/roi_pool.py -------------------------------------------------------------------------------- /code/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/models/__init__.py -------------------------------------------------------------------------------- /code/models/midn_oicr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/models/midn_oicr.py -------------------------------------------------------------------------------- /code/models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/models/vgg16.py -------------------------------------------------------------------------------- /code/tasks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/tasks/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/tasks/config.py -------------------------------------------------------------------------------- /code/tasks/init_paths.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/tasks/init_paths.py -------------------------------------------------------------------------------- /code/tasks/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/tasks/test.py -------------------------------------------------------------------------------- /code/tasks/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/tasks/train.py -------------------------------------------------------------------------------- /code/tasks/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/tasks/visualize.py -------------------------------------------------------------------------------- /code/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/boxes.py -------------------------------------------------------------------------------- /code/utils/data_samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/data_samplers.py -------------------------------------------------------------------------------- /code/utils/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/data_transforms.py -------------------------------------------------------------------------------- /code/utils/detectron_weight_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/detectron_weight_helper.py -------------------------------------------------------------------------------- /code/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/logging.py -------------------------------------------------------------------------------- /code/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/misc.py -------------------------------------------------------------------------------- /code/utils/optimizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/optimizer_utils.py -------------------------------------------------------------------------------- /code/utils/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/test_utils.py -------------------------------------------------------------------------------- /code/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/timer.py -------------------------------------------------------------------------------- /code/utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/training_stats.py -------------------------------------------------------------------------------- /code/utils/vgg_weights_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/code/utils/vgg_weights_helper.py -------------------------------------------------------------------------------- /configs/baselines/vgg16_voc2007.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/configs/baselines/vgg16_voc2007.yaml -------------------------------------------------------------------------------- /configs/baselines/vgg16_voc2012.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/configs/baselines/vgg16_voc2012.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CatOneTwo/OICR-PyTorch/HEAD/requirements.txt --------------------------------------------------------------------------------