├── .github ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── ISSUE_TEMPLATE │ ├── Bugs.md │ ├── Feature-request.md │ └── documentation.md └── workflows │ └── workflows.yaml ├── .gitignore ├── LICENSE ├── MODEL_ZOO.md ├── README.md ├── configs ├── faster_rcnn_fbnetv3a_C4.yaml ├── faster_rcnn_fbnetv3a_C4_FSDP.yaml ├── faster_rcnn_fbnetv3a_C4_LSJ.yaml ├── faster_rcnn_fbnetv3a_dsmask_C4.yaml ├── faster_rcnn_fbnetv3g_fpn.yaml ├── keypoint_rcnn_fbnetv3a_dsmask_C4.yaml ├── mask_rcnn_fbnetv3a_C4.yaml ├── mask_rcnn_fbnetv3a_dsmask_C4.yaml ├── mask_rcnn_fbnetv3g_fpn.yaml └── qat_faster_rcnn_fbnetv3a_C4.yaml ├── d2go ├── __init__.py ├── checkpoint │ ├── __init__.py │ ├── api.py │ ├── checkpoint_instrumentation.py │ ├── fsdp_checkpoint.py │ ├── log_checkpoint.py │ └── utils.py ├── config │ ├── __init__.py │ ├── config.py │ └── utils.py ├── data │ ├── __init__.py │ ├── build.py │ ├── cache_util.py │ ├── config.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── build.py │ │ ├── d2go_dataset_mapper.py │ │ ├── data_reading.py │ │ └── rotated_dataset_mapper.py │ ├── datasets.py │ ├── disk_cache.py │ ├── extended_coco.py │ ├── extended_lvis.py │ ├── keypoint_metadata_registry.py │ ├── transforms │ │ ├── __init__.py │ │ ├── affine.py │ │ ├── auto_aug.py │ │ ├── blur.py │ │ ├── box_utils.py │ │ ├── build.py │ │ ├── color_yuv.py │ │ ├── crop.py │ │ ├── d2_native.py │ │ └── tensor.py │ └── utils.py ├── distributed.py ├── evaluation │ ├── __init__.py │ ├── api.py │ ├── evaluator.py │ ├── prediction_count_evaluation.py │ └── sem_seg_evaluation.py ├── export │ ├── __init__.py │ ├── api.py │ ├── exporter.py │ └── torchscript.py ├── initializer.py ├── model_zoo │ ├── __init__.py │ └── model_zoo.py ├── modeling │ ├── __init__.py │ ├── api.py │ ├── backbone │ │ ├── __init__.py │ │ ├── fbnet_cfg.py │ │ ├── fbnet_v2.py │ │ └── modules.py │ ├── distillation.py │ ├── ema.py │ ├── image_pooler.py │ ├── kmeans_anchors.py │ ├── meta_arch │ │ ├── __init__.py │ │ ├── fcos.py │ │ ├── panoptic_fpn.py │ │ ├── rcnn.py │ │ ├── retinanet.py │ │ └── semantic_seg.py │ ├── misc.py │ ├── model_freezing_utils.py │ ├── modeldef │ │ ├── __init__.py │ │ ├── fbnet_modeldef_registry.py │ │ └── modeldef.py │ ├── modeling_hook.py │ └── subclass.py ├── optimizer │ ├── __init__.py │ └── build.py ├── quantization │ ├── __init__.py │ ├── fx.py │ ├── learnable_qat.py │ ├── modeling.py │ └── qconfig.py ├── registry │ ├── __init__.py │ ├── bootstrap.py │ └── builtin.py ├── runner │ ├── __init__.py │ ├── api.py │ ├── callbacks │ │ ├── __init__.py │ │ └── quantization.py │ ├── config_defaults.py │ ├── debug_runner.py │ ├── default_runner.py │ ├── lightning_task.py │ └── training_hooks.py ├── setup.py ├── trainer │ ├── __init__.py │ ├── activation_checkpointing.py │ ├── api.py │ ├── fsdp.py │ ├── helper.py │ └── lightning │ │ ├── __init__.py │ │ └── training_loop.py └── utils │ ├── __init__.py │ ├── abnormal_checker.py │ ├── demo_predictor.py │ ├── ema_state.py │ ├── flop_calculator.py │ ├── gpu_memory_profiler.py │ ├── helper.py │ ├── launch_environment.py │ ├── logging.py │ ├── mast.py │ ├── misc.py │ ├── parse_module_params.py │ ├── tensorboard_log_util.py │ ├── testing │ ├── __init__.py │ ├── data_loader_helper.py │ ├── helper.py │ ├── lightning_test_module.py │ ├── lightning_train_helper.py │ ├── meta_arch_helper.py │ ├── rcnn_helper.py │ └── sem_seg_helper.py │ ├── validation_monitor.py │ └── visualization.py ├── demo ├── README.md ├── d2go_beginner.ipynb └── demo.py ├── projects_oss └── detr │ ├── LICENSE │ ├── README.md │ ├── configs │ ├── deformable_detr_fbnetv3a_bs16.yaml │ └── detr_fbnetv3a_bs16.yaml │ ├── detr │ ├── __init__.py │ ├── backbone │ │ ├── deit.py │ │ └── pit.py │ ├── d2 │ │ ├── __init__.py │ │ ├── config.py │ │ ├── dataset_mapper.py │ │ └── detr.py │ ├── datasets │ │ ├── __init__.py │ │ ├── ade.py │ │ ├── coco.py │ │ ├── coco_eval.py │ │ ├── coco_panoptic.py │ │ ├── panoptic_eval.py │ │ └── transforms.py │ ├── engine.py │ ├── functions │ │ ├── __init__.py │ │ └── ms_deform_attn_func.py │ ├── hub.py │ ├── models │ │ ├── __init__.py │ │ ├── backbone.py │ │ ├── build.py │ │ ├── deformable_detr.py │ │ ├── deformable_transformer.py │ │ ├── detr.py │ │ ├── matcher.py │ │ ├── position_encoding.py │ │ ├── segmentation.py │ │ ├── setcriterion.py │ │ └── transformer.py │ ├── modules │ │ ├── __init__.py │ │ └── ms_deform_attn.py │ ├── runner.py │ ├── src │ │ ├── cpu │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ └── ms_deform_attn_cpu.h │ │ ├── gpu │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ ├── ms_deform_attn_cuda.h │ │ │ └── ms_deform_im2col_cuda.cuh │ │ ├── ms_deform_attn.h │ │ └── vision.cpp │ └── util │ │ ├── __init__.py │ │ ├── box_ops.py │ │ ├── misc.py │ │ └── plot_utils.py │ ├── main.py │ ├── setup.py │ ├── test_all.py │ ├── test_deit_backbone.py │ ├── test_detr_export.py │ ├── test_detr_runner.py │ └── test_op.py ├── setup.py ├── tests ├── __init__.py ├── data │ ├── __init__.py │ ├── test_d2go_datasets.py │ ├── test_data_loader.py │ ├── test_data_transforms.py │ ├── test_data_transforms_affine.py │ ├── test_data_transforms_auto_aug.py │ ├── test_data_transforms_blur.py │ ├── test_data_transforms_box_utils.py │ ├── test_data_transforms_color_yuv.py │ ├── test_data_transforms_crop.py │ └── test_data_transforms_tensor.py ├── evaluation │ ├── __init__.py │ ├── test_evaluator.py │ └── test_prediction_count_evaluation.py ├── export │ ├── __init__.py │ ├── test_api.py │ └── test_torchscript.py ├── misc │ ├── __init__.py │ ├── test_config.py │ ├── test_flop_count.py │ ├── test_lightning_train_net.py │ ├── test_registration.py │ ├── test_sub_package_init_file.py │ ├── test_utils_abnormal_checker.py │ └── test_validation_monitor.py ├── modeling │ ├── __init__.py │ ├── test_box2box_transform.py │ ├── test_box_with_nms_limit.py │ ├── test_kmeans_anchors.py │ ├── test_meta_arch_rcnn.py │ ├── test_meta_arch_semantic_seg.py │ ├── test_model_zoo.py │ ├── test_modeling_distillation.py │ ├── test_modeling_ema.py │ ├── test_modeling_image_pooler.py │ ├── test_modeling_meta_arch_modeling_hook.py │ ├── test_nms.py │ ├── test_optimizer.py │ ├── test_rcnn_export_example.py │ ├── test_rcnn_helper.py │ └── test_rpn_heads.py ├── resources │ ├── arch_def_merging.yaml │ ├── defaults_gen_case1.yaml │ ├── diff_aug_rand_crop.png │ ├── diff_aug_rand_cutout.png │ ├── diff_aug_rand_translation.png │ ├── rerouted_base.yaml │ └── rerouted_multi_base.yaml ├── runner │ ├── __init__.py │ ├── test_runner_default_runner.py │ ├── test_runner_lightning_quantization.py │ └── test_runner_lightning_task.py ├── skip_init │ ├── __init__.py │ └── test_registries_bootstrap.py ├── trainer │ ├── __init__.py │ └── test_activation_checkpointing.py └── utils │ ├── __init__.py │ ├── test_gpu_memory_profiler.py │ └── test_visualization.py └── tools ├── __init__.py ├── benchmark_data.py ├── evaluator.py ├── exporter.py ├── lightning_train_net.py └── train_net.py /.github/CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/Bugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/ISSUE_TEMPLATE/Bugs.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/Feature-request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/ISSUE_TEMPLATE/Feature-request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/documentation.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/ISSUE_TEMPLATE/documentation.md -------------------------------------------------------------------------------- /.github/workflows/workflows.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.github/workflows/workflows.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/LICENSE -------------------------------------------------------------------------------- /MODEL_ZOO.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/MODEL_ZOO.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/README.md -------------------------------------------------------------------------------- /configs/faster_rcnn_fbnetv3a_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/faster_rcnn_fbnetv3a_C4.yaml -------------------------------------------------------------------------------- /configs/faster_rcnn_fbnetv3a_C4_FSDP.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/faster_rcnn_fbnetv3a_C4_FSDP.yaml -------------------------------------------------------------------------------- /configs/faster_rcnn_fbnetv3a_C4_LSJ.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/faster_rcnn_fbnetv3a_C4_LSJ.yaml -------------------------------------------------------------------------------- /configs/faster_rcnn_fbnetv3a_dsmask_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/faster_rcnn_fbnetv3a_dsmask_C4.yaml -------------------------------------------------------------------------------- /configs/faster_rcnn_fbnetv3g_fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/faster_rcnn_fbnetv3g_fpn.yaml -------------------------------------------------------------------------------- /configs/keypoint_rcnn_fbnetv3a_dsmask_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/keypoint_rcnn_fbnetv3a_dsmask_C4.yaml -------------------------------------------------------------------------------- /configs/mask_rcnn_fbnetv3a_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/mask_rcnn_fbnetv3a_C4.yaml -------------------------------------------------------------------------------- /configs/mask_rcnn_fbnetv3a_dsmask_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/mask_rcnn_fbnetv3a_dsmask_C4.yaml -------------------------------------------------------------------------------- /configs/mask_rcnn_fbnetv3g_fpn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/mask_rcnn_fbnetv3g_fpn.yaml -------------------------------------------------------------------------------- /configs/qat_faster_rcnn_fbnetv3a_C4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/configs/qat_faster_rcnn_fbnetv3a_C4.yaml -------------------------------------------------------------------------------- /d2go/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/__init__.py -------------------------------------------------------------------------------- /d2go/checkpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/__init__.py -------------------------------------------------------------------------------- /d2go/checkpoint/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/api.py -------------------------------------------------------------------------------- /d2go/checkpoint/checkpoint_instrumentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/checkpoint_instrumentation.py -------------------------------------------------------------------------------- /d2go/checkpoint/fsdp_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/fsdp_checkpoint.py -------------------------------------------------------------------------------- /d2go/checkpoint/log_checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/log_checkpoint.py -------------------------------------------------------------------------------- /d2go/checkpoint/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/checkpoint/utils.py -------------------------------------------------------------------------------- /d2go/config/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/config/__init__.py -------------------------------------------------------------------------------- /d2go/config/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/config/config.py -------------------------------------------------------------------------------- /d2go/config/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/config/utils.py -------------------------------------------------------------------------------- /d2go/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/__init__.py -------------------------------------------------------------------------------- /d2go/data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/build.py -------------------------------------------------------------------------------- /d2go/data/cache_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/cache_util.py -------------------------------------------------------------------------------- /d2go/data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/config.py -------------------------------------------------------------------------------- /d2go/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/dataset_mappers/__init__.py -------------------------------------------------------------------------------- /d2go/data/dataset_mappers/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/dataset_mappers/build.py -------------------------------------------------------------------------------- /d2go/data/dataset_mappers/d2go_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/dataset_mappers/d2go_dataset_mapper.py -------------------------------------------------------------------------------- /d2go/data/dataset_mappers/data_reading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/dataset_mappers/data_reading.py -------------------------------------------------------------------------------- /d2go/data/dataset_mappers/rotated_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/dataset_mappers/rotated_dataset_mapper.py -------------------------------------------------------------------------------- /d2go/data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/datasets.py -------------------------------------------------------------------------------- /d2go/data/disk_cache.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/disk_cache.py -------------------------------------------------------------------------------- /d2go/data/extended_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/extended_coco.py -------------------------------------------------------------------------------- /d2go/data/extended_lvis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/extended_lvis.py -------------------------------------------------------------------------------- /d2go/data/keypoint_metadata_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/keypoint_metadata_registry.py -------------------------------------------------------------------------------- /d2go/data/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/__init__.py -------------------------------------------------------------------------------- /d2go/data/transforms/affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/affine.py -------------------------------------------------------------------------------- /d2go/data/transforms/auto_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/auto_aug.py -------------------------------------------------------------------------------- /d2go/data/transforms/blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/blur.py -------------------------------------------------------------------------------- /d2go/data/transforms/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/box_utils.py -------------------------------------------------------------------------------- /d2go/data/transforms/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/build.py -------------------------------------------------------------------------------- /d2go/data/transforms/color_yuv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/color_yuv.py -------------------------------------------------------------------------------- /d2go/data/transforms/crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/crop.py -------------------------------------------------------------------------------- /d2go/data/transforms/d2_native.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/d2_native.py -------------------------------------------------------------------------------- /d2go/data/transforms/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/transforms/tensor.py -------------------------------------------------------------------------------- /d2go/data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/data/utils.py -------------------------------------------------------------------------------- /d2go/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/distributed.py -------------------------------------------------------------------------------- /d2go/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/evaluation/__init__.py -------------------------------------------------------------------------------- /d2go/evaluation/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/evaluation/api.py -------------------------------------------------------------------------------- /d2go/evaluation/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/evaluation/evaluator.py -------------------------------------------------------------------------------- /d2go/evaluation/prediction_count_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/evaluation/prediction_count_evaluation.py -------------------------------------------------------------------------------- /d2go/evaluation/sem_seg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/evaluation/sem_seg_evaluation.py -------------------------------------------------------------------------------- /d2go/export/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/export/__init__.py -------------------------------------------------------------------------------- /d2go/export/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/export/api.py -------------------------------------------------------------------------------- /d2go/export/exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/export/exporter.py -------------------------------------------------------------------------------- /d2go/export/torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/export/torchscript.py -------------------------------------------------------------------------------- /d2go/initializer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/initializer.py -------------------------------------------------------------------------------- /d2go/model_zoo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /d2go/model_zoo/model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/model_zoo/model_zoo.py -------------------------------------------------------------------------------- /d2go/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/__init__.py -------------------------------------------------------------------------------- /d2go/modeling/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/api.py -------------------------------------------------------------------------------- /d2go/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/backbone/__init__.py -------------------------------------------------------------------------------- /d2go/modeling/backbone/fbnet_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/backbone/fbnet_cfg.py -------------------------------------------------------------------------------- /d2go/modeling/backbone/fbnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/backbone/fbnet_v2.py -------------------------------------------------------------------------------- /d2go/modeling/backbone/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/backbone/modules.py -------------------------------------------------------------------------------- /d2go/modeling/distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/distillation.py -------------------------------------------------------------------------------- /d2go/modeling/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/ema.py -------------------------------------------------------------------------------- /d2go/modeling/image_pooler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/image_pooler.py -------------------------------------------------------------------------------- /d2go/modeling/kmeans_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/kmeans_anchors.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/__init__.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/fcos.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/fcos.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/panoptic_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/panoptic_fpn.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/rcnn.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/retinanet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/retinanet.py -------------------------------------------------------------------------------- /d2go/modeling/meta_arch/semantic_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/meta_arch/semantic_seg.py -------------------------------------------------------------------------------- /d2go/modeling/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/misc.py -------------------------------------------------------------------------------- /d2go/modeling/model_freezing_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/model_freezing_utils.py -------------------------------------------------------------------------------- /d2go/modeling/modeldef/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/modeldef/__init__.py -------------------------------------------------------------------------------- /d2go/modeling/modeldef/fbnet_modeldef_registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/modeldef/fbnet_modeldef_registry.py -------------------------------------------------------------------------------- /d2go/modeling/modeldef/modeldef.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/modeldef/modeldef.py -------------------------------------------------------------------------------- /d2go/modeling/modeling_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/modeling_hook.py -------------------------------------------------------------------------------- /d2go/modeling/subclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/modeling/subclass.py -------------------------------------------------------------------------------- /d2go/optimizer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/optimizer/__init__.py -------------------------------------------------------------------------------- /d2go/optimizer/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/optimizer/build.py -------------------------------------------------------------------------------- /d2go/quantization/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /d2go/quantization/fx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/quantization/fx.py -------------------------------------------------------------------------------- /d2go/quantization/learnable_qat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/quantization/learnable_qat.py -------------------------------------------------------------------------------- /d2go/quantization/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/quantization/modeling.py -------------------------------------------------------------------------------- /d2go/quantization/qconfig.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/quantization/qconfig.py -------------------------------------------------------------------------------- /d2go/registry/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/registry/__init__.py -------------------------------------------------------------------------------- /d2go/registry/bootstrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/registry/bootstrap.py -------------------------------------------------------------------------------- /d2go/registry/builtin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/registry/builtin.py -------------------------------------------------------------------------------- /d2go/runner/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/__init__.py -------------------------------------------------------------------------------- /d2go/runner/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/api.py -------------------------------------------------------------------------------- /d2go/runner/callbacks/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /d2go/runner/callbacks/quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/callbacks/quantization.py -------------------------------------------------------------------------------- /d2go/runner/config_defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/config_defaults.py -------------------------------------------------------------------------------- /d2go/runner/debug_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/debug_runner.py -------------------------------------------------------------------------------- /d2go/runner/default_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/default_runner.py -------------------------------------------------------------------------------- /d2go/runner/lightning_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/lightning_task.py -------------------------------------------------------------------------------- /d2go/runner/training_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/runner/training_hooks.py -------------------------------------------------------------------------------- /d2go/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/setup.py -------------------------------------------------------------------------------- /d2go/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/__init__.py -------------------------------------------------------------------------------- /d2go/trainer/activation_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/activation_checkpointing.py -------------------------------------------------------------------------------- /d2go/trainer/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/api.py -------------------------------------------------------------------------------- /d2go/trainer/fsdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/fsdp.py -------------------------------------------------------------------------------- /d2go/trainer/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/helper.py -------------------------------------------------------------------------------- /d2go/trainer/lightning/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/lightning/__init__.py -------------------------------------------------------------------------------- /d2go/trainer/lightning/training_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/trainer/lightning/training_loop.py -------------------------------------------------------------------------------- /d2go/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/__init__.py -------------------------------------------------------------------------------- /d2go/utils/abnormal_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/abnormal_checker.py -------------------------------------------------------------------------------- /d2go/utils/demo_predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/demo_predictor.py -------------------------------------------------------------------------------- /d2go/utils/ema_state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/ema_state.py -------------------------------------------------------------------------------- /d2go/utils/flop_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/flop_calculator.py -------------------------------------------------------------------------------- /d2go/utils/gpu_memory_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/gpu_memory_profiler.py -------------------------------------------------------------------------------- /d2go/utils/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/helper.py -------------------------------------------------------------------------------- /d2go/utils/launch_environment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/launch_environment.py -------------------------------------------------------------------------------- /d2go/utils/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/logging.py -------------------------------------------------------------------------------- /d2go/utils/mast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/mast.py -------------------------------------------------------------------------------- /d2go/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/misc.py -------------------------------------------------------------------------------- /d2go/utils/parse_module_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/parse_module_params.py -------------------------------------------------------------------------------- /d2go/utils/tensorboard_log_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/tensorboard_log_util.py -------------------------------------------------------------------------------- /d2go/utils/testing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/__init__.py -------------------------------------------------------------------------------- /d2go/utils/testing/data_loader_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/data_loader_helper.py -------------------------------------------------------------------------------- /d2go/utils/testing/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/helper.py -------------------------------------------------------------------------------- /d2go/utils/testing/lightning_test_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/lightning_test_module.py -------------------------------------------------------------------------------- /d2go/utils/testing/lightning_train_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/lightning_train_helper.py -------------------------------------------------------------------------------- /d2go/utils/testing/meta_arch_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/meta_arch_helper.py -------------------------------------------------------------------------------- /d2go/utils/testing/rcnn_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/rcnn_helper.py -------------------------------------------------------------------------------- /d2go/utils/testing/sem_seg_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/testing/sem_seg_helper.py -------------------------------------------------------------------------------- /d2go/utils/validation_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/validation_monitor.py -------------------------------------------------------------------------------- /d2go/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/d2go/utils/visualization.py -------------------------------------------------------------------------------- /demo/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/demo/README.md -------------------------------------------------------------------------------- /demo/d2go_beginner.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/demo/d2go_beginner.ipynb -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/demo/demo.py -------------------------------------------------------------------------------- /projects_oss/detr/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/LICENSE -------------------------------------------------------------------------------- /projects_oss/detr/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/README.md -------------------------------------------------------------------------------- /projects_oss/detr/configs/deformable_detr_fbnetv3a_bs16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/configs/deformable_detr_fbnetv3a_bs16.yaml -------------------------------------------------------------------------------- /projects_oss/detr/configs/detr_fbnetv3a_bs16.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/configs/detr_fbnetv3a_bs16.yaml -------------------------------------------------------------------------------- /projects_oss/detr/detr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/backbone/deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/backbone/deit.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/backbone/pit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/backbone/pit.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/d2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/d2/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/d2/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/d2/config.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/d2/dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/d2/dataset_mapper.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/d2/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/d2/detr.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/ade.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/ade.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/coco.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/coco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/coco_eval.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/coco_panoptic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/coco_panoptic.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/panoptic_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/panoptic_eval.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/datasets/transforms.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/engine.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/functions/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/hub.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/backbone.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/build.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/deformable_detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/deformable_detr.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/deformable_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/deformable_transformer.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/detr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/detr.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/matcher.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/position_encoding.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/segmentation.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/setcriterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/setcriterion.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/models/transformer.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/modules/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/runner.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/gpu/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/gpu/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/gpu/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/gpu/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/gpu/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/gpu/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/ms_deform_attn.h -------------------------------------------------------------------------------- /projects_oss/detr/detr/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/src/vision.cpp -------------------------------------------------------------------------------- /projects_oss/detr/detr/util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/util/__init__.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/util/box_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/util/box_ops.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/util/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/util/misc.py -------------------------------------------------------------------------------- /projects_oss/detr/detr/util/plot_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/detr/util/plot_utils.py -------------------------------------------------------------------------------- /projects_oss/detr/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/main.py -------------------------------------------------------------------------------- /projects_oss/detr/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/setup.py -------------------------------------------------------------------------------- /projects_oss/detr/test_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/test_all.py -------------------------------------------------------------------------------- /projects_oss/detr/test_deit_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/test_deit_backbone.py -------------------------------------------------------------------------------- /projects_oss/detr/test_detr_export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/test_detr_export.py -------------------------------------------------------------------------------- /projects_oss/detr/test_detr_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/test_detr_runner.py -------------------------------------------------------------------------------- /projects_oss/detr/test_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/projects_oss/detr/test_op.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/data/test_d2go_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_d2go_datasets.py -------------------------------------------------------------------------------- /tests/data/test_data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_loader.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_affine.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_auto_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_auto_aug.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_blur.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_blur.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_box_utils.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_color_yuv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_color_yuv.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_crop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_crop.py -------------------------------------------------------------------------------- /tests/data/test_data_transforms_tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/data/test_data_transforms_tensor.py -------------------------------------------------------------------------------- /tests/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/evaluation/test_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/evaluation/test_evaluator.py -------------------------------------------------------------------------------- /tests/evaluation/test_prediction_count_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/evaluation/test_prediction_count_evaluation.py -------------------------------------------------------------------------------- /tests/export/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/export/test_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/export/test_api.py -------------------------------------------------------------------------------- /tests/export/test_torchscript.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/export/test_torchscript.py -------------------------------------------------------------------------------- /tests/misc/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/misc/test_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_config.py -------------------------------------------------------------------------------- /tests/misc/test_flop_count.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_flop_count.py -------------------------------------------------------------------------------- /tests/misc/test_lightning_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_lightning_train_net.py -------------------------------------------------------------------------------- /tests/misc/test_registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_registration.py -------------------------------------------------------------------------------- /tests/misc/test_sub_package_init_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_sub_package_init_file.py -------------------------------------------------------------------------------- /tests/misc/test_utils_abnormal_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_utils_abnormal_checker.py -------------------------------------------------------------------------------- /tests/misc/test_validation_monitor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/misc/test_validation_monitor.py -------------------------------------------------------------------------------- /tests/modeling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeling/test_box2box_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_box2box_transform.py -------------------------------------------------------------------------------- /tests/modeling/test_box_with_nms_limit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_box_with_nms_limit.py -------------------------------------------------------------------------------- /tests/modeling/test_kmeans_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_kmeans_anchors.py -------------------------------------------------------------------------------- /tests/modeling/test_meta_arch_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_meta_arch_rcnn.py -------------------------------------------------------------------------------- /tests/modeling/test_meta_arch_semantic_seg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_meta_arch_semantic_seg.py -------------------------------------------------------------------------------- /tests/modeling/test_model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_model_zoo.py -------------------------------------------------------------------------------- /tests/modeling/test_modeling_distillation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_modeling_distillation.py -------------------------------------------------------------------------------- /tests/modeling/test_modeling_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_modeling_ema.py -------------------------------------------------------------------------------- /tests/modeling/test_modeling_image_pooler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_modeling_image_pooler.py -------------------------------------------------------------------------------- /tests/modeling/test_modeling_meta_arch_modeling_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_modeling_meta_arch_modeling_hook.py -------------------------------------------------------------------------------- /tests/modeling/test_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_nms.py -------------------------------------------------------------------------------- /tests/modeling/test_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_optimizer.py -------------------------------------------------------------------------------- /tests/modeling/test_rcnn_export_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_rcnn_export_example.py -------------------------------------------------------------------------------- /tests/modeling/test_rcnn_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_rcnn_helper.py -------------------------------------------------------------------------------- /tests/modeling/test_rpn_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/modeling/test_rpn_heads.py -------------------------------------------------------------------------------- /tests/resources/arch_def_merging.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/arch_def_merging.yaml -------------------------------------------------------------------------------- /tests/resources/defaults_gen_case1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/defaults_gen_case1.yaml -------------------------------------------------------------------------------- /tests/resources/diff_aug_rand_crop.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/diff_aug_rand_crop.png -------------------------------------------------------------------------------- /tests/resources/diff_aug_rand_cutout.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/diff_aug_rand_cutout.png -------------------------------------------------------------------------------- /tests/resources/diff_aug_rand_translation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/diff_aug_rand_translation.png -------------------------------------------------------------------------------- /tests/resources/rerouted_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/rerouted_base.yaml -------------------------------------------------------------------------------- /tests/resources/rerouted_multi_base.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/resources/rerouted_multi_base.yaml -------------------------------------------------------------------------------- /tests/runner/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/runner/test_runner_default_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/runner/test_runner_default_runner.py -------------------------------------------------------------------------------- /tests/runner/test_runner_lightning_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/runner/test_runner_lightning_quantization.py -------------------------------------------------------------------------------- /tests/runner/test_runner_lightning_task.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/runner/test_runner_lightning_task.py -------------------------------------------------------------------------------- /tests/skip_init/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/skip_init/test_registries_bootstrap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/skip_init/test_registries_bootstrap.py -------------------------------------------------------------------------------- /tests/trainer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/trainer/test_activation_checkpointing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/trainer/test_activation_checkpointing.py -------------------------------------------------------------------------------- /tests/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/utils/test_gpu_memory_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/utils/test_gpu_memory_profiler.py -------------------------------------------------------------------------------- /tests/utils/test_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tests/utils/test_visualization.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /tools/benchmark_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tools/benchmark_data.py -------------------------------------------------------------------------------- /tools/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tools/evaluator.py -------------------------------------------------------------------------------- /tools/exporter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tools/exporter.py -------------------------------------------------------------------------------- /tools/lightning_train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tools/lightning_train_net.py -------------------------------------------------------------------------------- /tools/train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/d2go/HEAD/tools/train_net.py --------------------------------------------------------------------------------