├── .github ├── FUNDING.yml └── issue_template.md ├── .gitignore ├── CMakeLists.txt ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── FAQ.md ├── GETTING_STARTED.md ├── INSTALL.md ├── LICENSE ├── MODEL_ZOO.md ├── Makefile ├── NOTICE ├── README.md ├── cmake ├── Summary.cmake └── legacy │ ├── Cuda.cmake │ ├── Dependencies.cmake │ ├── Modules │ └── FindCuDNN.cmake │ ├── Summary.cmake │ ├── Utils.cmake │ └── legacymake.cmake ├── configs ├── 04_2018_gn_baselines │ ├── e2e_mask_rcnn_R-101-FPN_2x_gn.yaml │ ├── e2e_mask_rcnn_R-101-FPN_3x_gn.yaml │ ├── e2e_mask_rcnn_R-50-FPN_2x_gn.yaml │ ├── e2e_mask_rcnn_R-50-FPN_3x_gn.yaml │ ├── mask_rcnn_R-50-FPN_1x_gn.yaml │ ├── scratch_e2e_mask_rcnn_R-101-FPN_3x_gn.yaml │ └── scratch_e2e_mask_rcnn_R-50-FPN_3x_gn.yaml ├── 12_2017_baselines │ ├── e2e_faster_rcnn_R-101-FPN_1x.yaml │ ├── e2e_faster_rcnn_R-101-FPN_2x.yaml │ ├── e2e_faster_rcnn_R-50-C4_1x.yaml │ ├── e2e_faster_rcnn_R-50-C4_2x.yaml │ ├── e2e_faster_rcnn_R-50-FPN_1x.yaml │ ├── e2e_faster_rcnn_R-50-FPN_270.yaml │ ├── e2e_faster_rcnn_R-50-FPN_2x.yaml │ ├── e2e_faster_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── e2e_faster_rcnn_X-101-32x8d-FPN_2x.yaml │ ├── e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── e2e_faster_rcnn_X-101-64x4d-FPN_2x.yaml │ ├── e2e_keypoint_rcnn_R-101-FPN_1x.yaml │ ├── e2e_keypoint_rcnn_R-101-FPN_s1x.yaml │ ├── e2e_keypoint_rcnn_R-50-FPN_1x.yaml │ ├── e2e_keypoint_rcnn_R-50-FPN_s1x.yaml │ ├── e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml │ ├── e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml │ ├── e2e_mask_rcnn_R-101-FPN_1x.yaml │ ├── e2e_mask_rcnn_R-101-FPN_2x.yaml │ ├── e2e_mask_rcnn_R-50-C4_1x.yaml │ ├── e2e_mask_rcnn_R-50-C4_2x.yaml │ ├── e2e_mask_rcnn_R-50-FPN_1x.yaml │ ├── e2e_mask_rcnn_R-50-FPN_2x.yaml │ ├── e2e_mask_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── e2e_mask_rcnn_X-101-32x8d-FPN_2x.yaml │ ├── e2e_mask_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── e2e_mask_rcnn_X-101-64x4d-FPN_2x.yaml │ ├── e2e_mask_rcnn_X-152-32x8d-FPN-IN5k_1.44x.yaml │ ├── fast_rcnn_R-101-FPN_1x.yaml │ ├── fast_rcnn_R-101-FPN_2x.yaml │ ├── fast_rcnn_R-50-C4_1x.yaml │ ├── fast_rcnn_R-50-C4_2x.yaml │ ├── fast_rcnn_R-50-FPN_1x.yaml │ ├── fast_rcnn_R-50-FPN_2x.yaml │ ├── fast_rcnn_R-50-FPN_2x_270.yaml │ ├── fast_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── fast_rcnn_X-101-32x8d-FPN_2x.yaml │ ├── fast_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── fast_rcnn_X-101-64x4d-FPN_2x.yaml │ ├── keypoint_rcnn_R-101-FPN_1x.yaml │ ├── keypoint_rcnn_R-101-FPN_s1x.yaml │ ├── keypoint_rcnn_R-50-FPN_1x.yaml │ ├── keypoint_rcnn_R-50-FPN_s1x.yaml │ ├── keypoint_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml │ ├── keypoint_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml │ ├── mask_rcnn_R-101-FPN_1x.yaml │ ├── mask_rcnn_R-101-FPN_2x.yaml │ ├── mask_rcnn_R-50-C4_1x.yaml │ ├── mask_rcnn_R-50-C4_2x.yaml │ ├── mask_rcnn_R-50-FPN_1x.yaml │ ├── mask_rcnn_R-50-FPN_2x.yaml │ ├── mask_rcnn_X-101-32x8d-FPN_1x.yaml │ ├── mask_rcnn_X-101-32x8d-FPN_2x.yaml │ ├── mask_rcnn_X-101-64x4d-FPN_1x.yaml │ ├── mask_rcnn_X-101-64x4d-FPN_2x.yaml │ ├── retinanet_R-101-FPN_1x.yaml │ ├── retinanet_R-101-FPN_2x.yaml │ ├── retinanet_R-50-FPN_1x.yaml │ ├── retinanet_R-50-FPN_2x.yaml │ ├── retinanet_X-101-32x8d-FPN_1x.yaml │ ├── retinanet_X-101-32x8d-FPN_2x.yaml │ ├── retinanet_X-101-64x4d-FPN_1x.yaml │ ├── retinanet_X-101-64x4d-FPN_2x.yaml │ ├── rpn_R-101-FPN_1x.yaml │ ├── rpn_R-50-C4_1x.yaml │ ├── rpn_R-50-FPN_1x.yaml │ ├── rpn_X-101-32x8d-FPN_1x.yaml │ ├── rpn_X-101-64x4d-FPN_1x.yaml │ ├── rpn_person_only_R-101-FPN_1x.yaml │ ├── rpn_person_only_R-50-FPN_1x.yaml │ ├── rpn_person_only_X-101-32x8d-FPN_1x.yaml │ └── rpn_person_only_X-101-64x4d-FPN_1x.yaml ├── e2e_faster_rcnn_R-50-FPN_2x.yaml ├── e2e_mask_rcnn_R-50-FPN_2x_log.yaml ├── getting_started │ ├── tutorial_1gpu_e2e_faster_rcnn_R-50-FPN.yaml │ ├── tutorial_2gpu_e2e_faster_rcnn_R-50-FPN.yaml │ ├── tutorial_4gpu_e2e_faster_rcnn_R-50-FPN.yaml │ └── tutorial_8gpu_e2e_faster_rcnn_R-50-FPN.yaml └── test_time_aug │ ├── e2e_mask_rcnn_R-50-FPN_2x.yaml │ └── keypoint_rcnn_R-50-FPN_1x.yaml ├── demo ├── 15673749081_767a7fa63a_k.jpg ├── 16004479832_a748d55f21_k.jpg ├── 17790319373_bd19b24cfc_k.jpg ├── 18124840932_e42b3e377c_k.jpg ├── 19064748793_bb942deea1_k.jpg ├── 24274813513_0cfd2ce6d0_k.jpg ├── 33823288584_1d21cf0a26_k.jpg ├── 33887522274_eebd074106_k.jpg ├── 34501842524_3c858b3080_k.jpg ├── NOTICE └── output │ ├── 17790319373_bd19b24cfc_k_example_output.jpg │ ├── 33823288584_1d21cf0a26_k_example_output.jpg │ └── softer.png ├── detectron ├── __init__.py ├── core │ ├── __init__.py │ ├── config.py │ ├── rpn_generator.py │ ├── test.py │ ├── test_engine.py │ └── test_retinanet.py ├── datasets │ ├── VOCdevkit-matlab-wrapper │ │ ├── get_voc_opts.m │ │ ├── voc_eval.m │ │ └── xVOCap.m │ ├── __init__.py │ ├── cityscapes_json_dataset_evaluator.py │ ├── coco_to_cityscapes_id.py │ ├── data │ │ └── README.md │ ├── dataset_catalog.py │ ├── dummy_datasets.py │ ├── json_dataset.py │ ├── json_dataset_evaluator.py │ ├── roidb.py │ ├── task_evaluation.py │ ├── voc_dataset_evaluator.py │ └── voc_eval.py ├── modeling │ ├── FPN.py │ ├── ResNet.py │ ├── VGG16.py │ ├── VGG_CNN_M_1024.py │ ├── __init__.py │ ├── detector.py │ ├── fast_rcnn_heads.py │ ├── generate_anchors.py │ ├── gradient_clipping.py │ ├── keypoint_rcnn_heads.py │ ├── mask_rcnn_heads.py │ ├── model_builder.py │ ├── name_compat.py │ ├── optimizer.py │ ├── retinanet_heads.py │ ├── rfcn_heads.py │ └── rpn_heads.py ├── ops │ ├── __init__.py │ ├── collect_and_distribute_fpn_rpn_proposals.py │ ├── generate_proposal_labels.py │ ├── generate_proposals.py │ ├── zero_even_op.cc │ ├── zero_even_op.cu │ └── zero_even_op.h ├── roi_data │ ├── __init__.py │ ├── data_utils.py │ ├── fast_rcnn.py │ ├── keypoint_rcnn.py │ ├── loader.py │ ├── mask_rcnn.py │ ├── minibatch.py │ ├── retinanet.py │ └── rpn.py ├── tests │ ├── data_loader_benchmark.py │ ├── test_batch_permutation_op.py │ ├── test_bbox_transform.py │ ├── test_cfg.py │ ├── test_loader.py │ ├── test_restore_checkpoint.py │ ├── test_smooth_l1_loss_op.py │ ├── test_spatial_narrow_as_op.py │ └── test_zero_even_op.py └── utils │ ├── __init__.py │ ├── blob.py │ ├── boxes.py │ ├── c2.py │ ├── collections.py │ ├── colormap.py │ ├── coordinator.py │ ├── cython_bbox.pyx │ ├── cython_nms.pyx │ ├── env.py │ ├── gmm.py │ ├── image.py │ ├── io.py │ ├── keypoints.py │ ├── logging.py │ ├── lr_policy.py │ ├── model_convert_utils.py │ ├── net.py │ ├── py_cpu_nms.py │ ├── segms.py │ ├── subprocess.py │ ├── timer.py │ ├── train.py │ ├── training_stats.py │ └── vis.py ├── docker └── Dockerfile ├── projects └── GN │ ├── README.md │ └── gn.jpg ├── requirements.txt ├── setup.py └── tools ├── convert_cityscapes_to_coco.py ├── convert_coco_model_to_cityscapes.py ├── convert_pkl_to_pb.py ├── convert_selective_search.py ├── generate_testdev_from_test.py ├── infer.py ├── infer_simple.py ├── pickle_caffe_blobs.py ├── read_log.py ├── reval.py ├── test_net.py ├── train_net.py └── visualize_results.py /.github/FUNDING.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/.github/FUNDING.yml -------------------------------------------------------------------------------- /.github/issue_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/.github/issue_template.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/.gitignore -------------------------------------------------------------------------------- /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /FAQ.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/FAQ.md -------------------------------------------------------------------------------- /GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/GETTING_STARTED.md -------------------------------------------------------------------------------- /INSTALL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/INSTALL.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/Makefile -------------------------------------------------------------------------------- /NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/NOTICE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/README.md -------------------------------------------------------------------------------- /cmake/Summary.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/Summary.cmake -------------------------------------------------------------------------------- /cmake/legacy/Cuda.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/Cuda.cmake -------------------------------------------------------------------------------- /cmake/legacy/Dependencies.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/Dependencies.cmake -------------------------------------------------------------------------------- /cmake/legacy/Modules/FindCuDNN.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/Modules/FindCuDNN.cmake -------------------------------------------------------------------------------- /cmake/legacy/Summary.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/Summary.cmake -------------------------------------------------------------------------------- /cmake/legacy/Utils.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/Utils.cmake -------------------------------------------------------------------------------- /cmake/legacy/legacymake.cmake: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/cmake/legacy/legacymake.cmake -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/e2e_mask_rcnn_R-101-FPN_2x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/e2e_mask_rcnn_R-101-FPN_2x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/e2e_mask_rcnn_R-101-FPN_3x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/e2e_mask_rcnn_R-101-FPN_3x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/e2e_mask_rcnn_R-50-FPN_2x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/e2e_mask_rcnn_R-50-FPN_2x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/e2e_mask_rcnn_R-50-FPN_3x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/e2e_mask_rcnn_R-50-FPN_3x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/mask_rcnn_R-50-FPN_1x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/mask_rcnn_R-50-FPN_1x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/scratch_e2e_mask_rcnn_R-101-FPN_3x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/scratch_e2e_mask_rcnn_R-101-FPN_3x_gn.yaml -------------------------------------------------------------------------------- /configs/04_2018_gn_baselines/scratch_e2e_mask_rcnn_R-50-FPN_3x_gn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/04_2018_gn_baselines/scratch_e2e_mask_rcnn_R-50-FPN_3x_gn.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-101-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-101-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-50-C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-50-C4_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-50-C4_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-50-C4_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_270.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_270.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_X-101-32x8d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_X-101-32x8d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_faster_rcnn_X-101-64x4d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_faster_rcnn_X-101-64x4d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_R-101-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_R-50-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_R-50-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-101-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-50-C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-50-C4_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-50-C4_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-50-C4_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_X-101-32x8d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_X-101-32x8d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_X-101-64x4d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_X-101-64x4d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/e2e_mask_rcnn_X-152-32x8d-FPN-IN5k_1.44x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/e2e_mask_rcnn_X-152-32x8d-FPN-IN5k_1.44x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-101-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-101-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-50-C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-50-C4_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-50-C4_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-50-C4_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_R-50-FPN_2x_270.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_R-50-FPN_2x_270.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_X-101-32x8d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_X-101-32x8d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/fast_rcnn_X-101-64x4d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/fast_rcnn_X-101-64x4d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_R-101-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_R-101-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_R-50-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_R-50-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_X-101-32x8d-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/keypoint_rcnn_X-101-64x4d-FPN_s1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-101-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-101-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-50-C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-50-C4_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-50-C4_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-50-C4_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_X-101-32x8d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_X-101-32x8d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/mask_rcnn_X-101-64x4d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/mask_rcnn_X-101-64x4d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_R-101-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_R-101-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_X-101-32x8d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_X-101-32x8d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/retinanet_X-101-64x4d-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/retinanet_X-101-64x4d-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_R-50-C4_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_R-50-C4_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_person_only_R-101-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_person_only_R-101-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_person_only_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_person_only_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_person_only_X-101-32x8d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_person_only_X-101-32x8d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/12_2017_baselines/rpn_person_only_X-101-64x4d-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/12_2017_baselines/rpn_person_only_X-101-64x4d-FPN_1x.yaml -------------------------------------------------------------------------------- /configs/e2e_faster_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/e2e_faster_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/e2e_mask_rcnn_R-50-FPN_2x_log.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/e2e_mask_rcnn_R-50-FPN_2x_log.yaml -------------------------------------------------------------------------------- /configs/getting_started/tutorial_1gpu_e2e_faster_rcnn_R-50-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/getting_started/tutorial_1gpu_e2e_faster_rcnn_R-50-FPN.yaml -------------------------------------------------------------------------------- /configs/getting_started/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/getting_started/tutorial_2gpu_e2e_faster_rcnn_R-50-FPN.yaml -------------------------------------------------------------------------------- /configs/getting_started/tutorial_4gpu_e2e_faster_rcnn_R-50-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/getting_started/tutorial_4gpu_e2e_faster_rcnn_R-50-FPN.yaml -------------------------------------------------------------------------------- /configs/getting_started/tutorial_8gpu_e2e_faster_rcnn_R-50-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/getting_started/tutorial_8gpu_e2e_faster_rcnn_R-50-FPN.yaml -------------------------------------------------------------------------------- /configs/test_time_aug/e2e_mask_rcnn_R-50-FPN_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/test_time_aug/e2e_mask_rcnn_R-50-FPN_2x.yaml -------------------------------------------------------------------------------- /configs/test_time_aug/keypoint_rcnn_R-50-FPN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/configs/test_time_aug/keypoint_rcnn_R-50-FPN_1x.yaml -------------------------------------------------------------------------------- /demo/15673749081_767a7fa63a_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/15673749081_767a7fa63a_k.jpg -------------------------------------------------------------------------------- /demo/16004479832_a748d55f21_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/16004479832_a748d55f21_k.jpg -------------------------------------------------------------------------------- /demo/17790319373_bd19b24cfc_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/17790319373_bd19b24cfc_k.jpg -------------------------------------------------------------------------------- /demo/18124840932_e42b3e377c_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/18124840932_e42b3e377c_k.jpg -------------------------------------------------------------------------------- /demo/19064748793_bb942deea1_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/19064748793_bb942deea1_k.jpg -------------------------------------------------------------------------------- /demo/24274813513_0cfd2ce6d0_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/24274813513_0cfd2ce6d0_k.jpg -------------------------------------------------------------------------------- /demo/33823288584_1d21cf0a26_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/33823288584_1d21cf0a26_k.jpg -------------------------------------------------------------------------------- /demo/33887522274_eebd074106_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/33887522274_eebd074106_k.jpg -------------------------------------------------------------------------------- /demo/34501842524_3c858b3080_k.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/34501842524_3c858b3080_k.jpg -------------------------------------------------------------------------------- /demo/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/NOTICE -------------------------------------------------------------------------------- /demo/output/17790319373_bd19b24cfc_k_example_output.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/output/17790319373_bd19b24cfc_k_example_output.jpg -------------------------------------------------------------------------------- /demo/output/33823288584_1d21cf0a26_k_example_output.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/output/33823288584_1d21cf0a26_k_example_output.jpg -------------------------------------------------------------------------------- /demo/output/softer.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/demo/output/softer.png -------------------------------------------------------------------------------- /detectron/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/core/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/core/config.py -------------------------------------------------------------------------------- /detectron/core/rpn_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/core/rpn_generator.py -------------------------------------------------------------------------------- /detectron/core/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/core/test.py -------------------------------------------------------------------------------- /detectron/core/test_engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/core/test_engine.py -------------------------------------------------------------------------------- /detectron/core/test_retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/core/test_retinanet.py -------------------------------------------------------------------------------- /detectron/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/VOCdevkit-matlab-wrapper/get_voc_opts.m -------------------------------------------------------------------------------- /detectron/datasets/VOCdevkit-matlab-wrapper/voc_eval.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/VOCdevkit-matlab-wrapper/voc_eval.m -------------------------------------------------------------------------------- /detectron/datasets/VOCdevkit-matlab-wrapper/xVOCap.m: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/VOCdevkit-matlab-wrapper/xVOCap.m -------------------------------------------------------------------------------- /detectron/datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/datasets/cityscapes_json_dataset_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/cityscapes_json_dataset_evaluator.py -------------------------------------------------------------------------------- /detectron/datasets/coco_to_cityscapes_id.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/coco_to_cityscapes_id.py -------------------------------------------------------------------------------- /detectron/datasets/data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/data/README.md -------------------------------------------------------------------------------- /detectron/datasets/dataset_catalog.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/dataset_catalog.py -------------------------------------------------------------------------------- /detectron/datasets/dummy_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/dummy_datasets.py -------------------------------------------------------------------------------- /detectron/datasets/json_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/json_dataset.py -------------------------------------------------------------------------------- /detectron/datasets/json_dataset_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/json_dataset_evaluator.py -------------------------------------------------------------------------------- /detectron/datasets/roidb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/roidb.py -------------------------------------------------------------------------------- /detectron/datasets/task_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/task_evaluation.py -------------------------------------------------------------------------------- /detectron/datasets/voc_dataset_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/voc_dataset_evaluator.py -------------------------------------------------------------------------------- /detectron/datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/datasets/voc_eval.py -------------------------------------------------------------------------------- /detectron/modeling/FPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/FPN.py -------------------------------------------------------------------------------- /detectron/modeling/ResNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/ResNet.py -------------------------------------------------------------------------------- /detectron/modeling/VGG16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/VGG16.py -------------------------------------------------------------------------------- /detectron/modeling/VGG_CNN_M_1024.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/VGG_CNN_M_1024.py -------------------------------------------------------------------------------- /detectron/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/__init__.py -------------------------------------------------------------------------------- /detectron/modeling/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/detector.py -------------------------------------------------------------------------------- /detectron/modeling/fast_rcnn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/fast_rcnn_heads.py -------------------------------------------------------------------------------- /detectron/modeling/generate_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/generate_anchors.py -------------------------------------------------------------------------------- /detectron/modeling/gradient_clipping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/gradient_clipping.py -------------------------------------------------------------------------------- /detectron/modeling/keypoint_rcnn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/keypoint_rcnn_heads.py -------------------------------------------------------------------------------- /detectron/modeling/mask_rcnn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/mask_rcnn_heads.py -------------------------------------------------------------------------------- /detectron/modeling/model_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/model_builder.py -------------------------------------------------------------------------------- /detectron/modeling/name_compat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/name_compat.py -------------------------------------------------------------------------------- /detectron/modeling/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/optimizer.py -------------------------------------------------------------------------------- /detectron/modeling/retinanet_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/retinanet_heads.py -------------------------------------------------------------------------------- /detectron/modeling/rfcn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/rfcn_heads.py -------------------------------------------------------------------------------- /detectron/modeling/rpn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/modeling/rpn_heads.py -------------------------------------------------------------------------------- /detectron/ops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/ops/collect_and_distribute_fpn_rpn_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/collect_and_distribute_fpn_rpn_proposals.py -------------------------------------------------------------------------------- /detectron/ops/generate_proposal_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/generate_proposal_labels.py -------------------------------------------------------------------------------- /detectron/ops/generate_proposals.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/generate_proposals.py -------------------------------------------------------------------------------- /detectron/ops/zero_even_op.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/zero_even_op.cc -------------------------------------------------------------------------------- /detectron/ops/zero_even_op.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/zero_even_op.cu -------------------------------------------------------------------------------- /detectron/ops/zero_even_op.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/ops/zero_even_op.h -------------------------------------------------------------------------------- /detectron/roi_data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/roi_data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/data_utils.py -------------------------------------------------------------------------------- /detectron/roi_data/fast_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/fast_rcnn.py -------------------------------------------------------------------------------- /detectron/roi_data/keypoint_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/keypoint_rcnn.py -------------------------------------------------------------------------------- /detectron/roi_data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/loader.py -------------------------------------------------------------------------------- /detectron/roi_data/mask_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/mask_rcnn.py -------------------------------------------------------------------------------- /detectron/roi_data/minibatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/minibatch.py -------------------------------------------------------------------------------- /detectron/roi_data/retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/retinanet.py -------------------------------------------------------------------------------- /detectron/roi_data/rpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/roi_data/rpn.py -------------------------------------------------------------------------------- /detectron/tests/data_loader_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/data_loader_benchmark.py -------------------------------------------------------------------------------- /detectron/tests/test_batch_permutation_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_batch_permutation_op.py -------------------------------------------------------------------------------- /detectron/tests/test_bbox_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_bbox_transform.py -------------------------------------------------------------------------------- /detectron/tests/test_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_cfg.py -------------------------------------------------------------------------------- /detectron/tests/test_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_loader.py -------------------------------------------------------------------------------- /detectron/tests/test_restore_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_restore_checkpoint.py -------------------------------------------------------------------------------- /detectron/tests/test_smooth_l1_loss_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_smooth_l1_loss_op.py -------------------------------------------------------------------------------- /detectron/tests/test_spatial_narrow_as_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_spatial_narrow_as_op.py -------------------------------------------------------------------------------- /detectron/tests/test_zero_even_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/tests/test_zero_even_op.py -------------------------------------------------------------------------------- /detectron/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /detectron/utils/blob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/blob.py -------------------------------------------------------------------------------- /detectron/utils/boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/boxes.py -------------------------------------------------------------------------------- /detectron/utils/c2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/c2.py -------------------------------------------------------------------------------- /detectron/utils/collections.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/collections.py -------------------------------------------------------------------------------- /detectron/utils/colormap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/colormap.py -------------------------------------------------------------------------------- /detectron/utils/coordinator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/coordinator.py -------------------------------------------------------------------------------- /detectron/utils/cython_bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/cython_bbox.pyx -------------------------------------------------------------------------------- /detectron/utils/cython_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/cython_nms.pyx -------------------------------------------------------------------------------- /detectron/utils/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/env.py -------------------------------------------------------------------------------- /detectron/utils/gmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/gmm.py -------------------------------------------------------------------------------- /detectron/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/image.py -------------------------------------------------------------------------------- /detectron/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/io.py -------------------------------------------------------------------------------- /detectron/utils/keypoints.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/keypoints.py -------------------------------------------------------------------------------- /detectron/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/logging.py -------------------------------------------------------------------------------- /detectron/utils/lr_policy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/lr_policy.py -------------------------------------------------------------------------------- /detectron/utils/model_convert_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/model_convert_utils.py -------------------------------------------------------------------------------- /detectron/utils/net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/net.py -------------------------------------------------------------------------------- /detectron/utils/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/py_cpu_nms.py -------------------------------------------------------------------------------- /detectron/utils/segms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/segms.py -------------------------------------------------------------------------------- /detectron/utils/subprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/subprocess.py -------------------------------------------------------------------------------- /detectron/utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/timer.py -------------------------------------------------------------------------------- /detectron/utils/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/train.py -------------------------------------------------------------------------------- /detectron/utils/training_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/training_stats.py -------------------------------------------------------------------------------- /detectron/utils/vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/detectron/utils/vis.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /projects/GN/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/projects/GN/README.md -------------------------------------------------------------------------------- /projects/GN/gn.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/projects/GN/gn.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/setup.py -------------------------------------------------------------------------------- /tools/convert_cityscapes_to_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/convert_cityscapes_to_coco.py -------------------------------------------------------------------------------- /tools/convert_coco_model_to_cityscapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/convert_coco_model_to_cityscapes.py -------------------------------------------------------------------------------- /tools/convert_pkl_to_pb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/convert_pkl_to_pb.py -------------------------------------------------------------------------------- /tools/convert_selective_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/convert_selective_search.py -------------------------------------------------------------------------------- /tools/generate_testdev_from_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/generate_testdev_from_test.py -------------------------------------------------------------------------------- /tools/infer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/infer.py -------------------------------------------------------------------------------- /tools/infer_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/infer_simple.py -------------------------------------------------------------------------------- /tools/pickle_caffe_blobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/pickle_caffe_blobs.py -------------------------------------------------------------------------------- /tools/read_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/read_log.py -------------------------------------------------------------------------------- /tools/reval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/reval.py -------------------------------------------------------------------------------- /tools/test_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/test_net.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/train_net.py -------------------------------------------------------------------------------- /tools/visualize_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ethanhe42/KL-Loss/HEAD/tools/visualize_results.py --------------------------------------------------------------------------------