├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── configs └── bua-caffe │ ├── extract-bua-caffe-r101-box-only.yaml │ ├── extract-bua-caffe-r101-fix36.yaml │ ├── extract-bua-caffe-r101-gt-bbox.yaml │ ├── extract-bua-caffe-r101.yaml │ ├── extract-bua-caffe-r152.yaml │ ├── test-bua-caffe-r101-fix36.yaml │ ├── test-bua-caffe-r101.yaml │ └── test-bua-caffe-r152.yaml ├── dataloader ├── __init__.py ├── dataset_mapper.py ├── dataset_vg.py ├── detection_utils.py ├── load_vg_json.py └── transform_gen.py ├── datasets ├── demo │ ├── 000456.jpg │ ├── 000542.jpg │ ├── 001150.jpg │ ├── 001763.jpg │ ├── 004545.jpg │ ├── example_image.jpg │ ├── example_image1.png │ └── example_image2.png └── demos │ ├── COCO_val2014_000000060623.jpg │ ├── COCO_val2014_000000115070.jpg │ ├── COCO_val2014_000000199962.jpg │ ├── COCO_val2014_000000201732.jpg │ ├── COCO_val2014_000000223648.jpg │ ├── COCO_val2014_000000384213.jpg │ ├── COCO_val2014_000000386164.jpg │ ├── COCO_val2014_000000391895.jpg │ ├── COCO_val2014_000000403385.jpg │ └── COCO_val2014_000000483108.jpg ├── evaluation ├── __init__.py ├── attributes_vocab.txt ├── objects_vocab.txt ├── vg_eval.py └── vg_evaluation.py ├── extract_features.py ├── models ├── __init__.py └── bua │ ├── __init__.py │ ├── backbone.py │ ├── box_regression.py │ ├── config.py │ ├── fast_rcnn.py │ ├── layers │ ├── csrc │ │ ├── __init__.py │ │ ├── nms │ │ │ ├── nms.cu │ │ │ ├── nms.h │ │ │ ├── nms_cpu.cpp │ │ │ ├── vision_cpu.h │ │ │ └── vision_cuda.h │ │ └── vision.cpp │ ├── nms.py │ └── wrappers.py │ ├── postprocessing.py │ ├── rcnn.py │ ├── roi_heads.py │ ├── rpn.py │ └── rpn_outputs.py ├── new_extract_features.py ├── opts.py ├── setup.py ├── train_net.py └── utils ├── __init__.py ├── extract_utils.py ├── extractor.py ├── utils.py └── visualize.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/README.md -------------------------------------------------------------------------------- /configs/bua-caffe/extract-bua-caffe-r101-box-only.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/extract-bua-caffe-r101-box-only.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/extract-bua-caffe-r101-fix36.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/extract-bua-caffe-r101-fix36.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/extract-bua-caffe-r101-gt-bbox.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/extract-bua-caffe-r101-gt-bbox.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/extract-bua-caffe-r101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/extract-bua-caffe-r101.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/extract-bua-caffe-r152.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/extract-bua-caffe-r152.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/test-bua-caffe-r101-fix36.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/test-bua-caffe-r101-fix36.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/test-bua-caffe-r101.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/test-bua-caffe-r101.yaml -------------------------------------------------------------------------------- /configs/bua-caffe/test-bua-caffe-r152.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/configs/bua-caffe/test-bua-caffe-r152.yaml -------------------------------------------------------------------------------- /dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/__init__.py -------------------------------------------------------------------------------- /dataloader/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/dataset_mapper.py -------------------------------------------------------------------------------- /dataloader/dataset_vg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/dataset_vg.py -------------------------------------------------------------------------------- /dataloader/detection_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/detection_utils.py -------------------------------------------------------------------------------- /dataloader/load_vg_json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/load_vg_json.py -------------------------------------------------------------------------------- /dataloader/transform_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/dataloader/transform_gen.py -------------------------------------------------------------------------------- /datasets/demo/000456.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/000456.jpg -------------------------------------------------------------------------------- /datasets/demo/000542.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/000542.jpg -------------------------------------------------------------------------------- /datasets/demo/001150.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/001150.jpg -------------------------------------------------------------------------------- /datasets/demo/001763.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/001763.jpg -------------------------------------------------------------------------------- /datasets/demo/004545.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/004545.jpg -------------------------------------------------------------------------------- /datasets/demo/example_image.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/example_image.jpg -------------------------------------------------------------------------------- /datasets/demo/example_image1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/example_image1.png -------------------------------------------------------------------------------- /datasets/demo/example_image2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demo/example_image2.png -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000060623.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000060623.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000115070.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000115070.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000199962.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000199962.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000201732.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000201732.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000223648.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000223648.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000384213.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000384213.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000386164.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000386164.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000391895.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000391895.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000403385.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000403385.jpg -------------------------------------------------------------------------------- /datasets/demos/COCO_val2014_000000483108.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/datasets/demos/COCO_val2014_000000483108.jpg -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/evaluation/__init__.py -------------------------------------------------------------------------------- /evaluation/attributes_vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/evaluation/attributes_vocab.txt -------------------------------------------------------------------------------- /evaluation/objects_vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/evaluation/objects_vocab.txt -------------------------------------------------------------------------------- /evaluation/vg_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/evaluation/vg_eval.py -------------------------------------------------------------------------------- /evaluation/vg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/evaluation/vg_evaluation.py -------------------------------------------------------------------------------- /extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/extract_features.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/bua/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/__init__.py -------------------------------------------------------------------------------- /models/bua/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/backbone.py -------------------------------------------------------------------------------- /models/bua/box_regression.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/box_regression.py -------------------------------------------------------------------------------- /models/bua/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/config.py -------------------------------------------------------------------------------- /models/bua/fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/fast_rcnn.py -------------------------------------------------------------------------------- /models/bua/layers/csrc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/__init__.py -------------------------------------------------------------------------------- /models/bua/layers/csrc/nms/nms.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/nms/nms.cu -------------------------------------------------------------------------------- /models/bua/layers/csrc/nms/nms.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/nms/nms.h -------------------------------------------------------------------------------- /models/bua/layers/csrc/nms/nms_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/nms/nms_cpu.cpp -------------------------------------------------------------------------------- /models/bua/layers/csrc/nms/vision_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/nms/vision_cpu.h -------------------------------------------------------------------------------- /models/bua/layers/csrc/nms/vision_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/nms/vision_cuda.h -------------------------------------------------------------------------------- /models/bua/layers/csrc/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/csrc/vision.cpp -------------------------------------------------------------------------------- /models/bua/layers/nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/nms.py -------------------------------------------------------------------------------- /models/bua/layers/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/layers/wrappers.py -------------------------------------------------------------------------------- /models/bua/postprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/postprocessing.py -------------------------------------------------------------------------------- /models/bua/rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/rcnn.py -------------------------------------------------------------------------------- /models/bua/roi_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/roi_heads.py -------------------------------------------------------------------------------- /models/bua/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/rpn.py -------------------------------------------------------------------------------- /models/bua/rpn_outputs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/models/bua/rpn_outputs.py -------------------------------------------------------------------------------- /new_extract_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/new_extract_features.py -------------------------------------------------------------------------------- /opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/opts.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/setup.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/train_net.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import save_features -------------------------------------------------------------------------------- /utils/extract_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/utils/extract_utils.py -------------------------------------------------------------------------------- /utils/extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/utils/extractor.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/visualize.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ezeli/bottom_up_features_extract/HEAD/utils/visualize.ipynb --------------------------------------------------------------------------------