path	# lines of code
pytorch/train.py	107
pytorch/launch_ddp.py	33
pytorch/sagemakercv/__init__.py	1
pytorch/sagemakercv/detection/backbone/__init__.py	1
pytorch/sagemakercv/detection/backbone/resnet.py	349
pytorch/sagemakercv/detection/backbone/fpn.py	77
pytorch/sagemakercv/detection/backbone/backbone.py	66
pytorch/sagemakercv/detection/registry.py	7
pytorch/sagemakercv/detection/detector/__init__.py	1
pytorch/sagemakercv/detection/detector/detectors.py	5
pytorch/sagemakercv/detection/detector/generalized_rcnn.py	58
pytorch/sagemakercv/detection/__init__.py	1
pytorch/sagemakercv/detection/rpn/__init__.py	1
pytorch/sagemakercv/detection/rpn/inference.py	305
pytorch/sagemakercv/detection/rpn/utils.py	28
pytorch/sagemakercv/detection/rpn/rpn.py	94
pytorch/sagemakercv/detection/rpn/anchor_generator.py	234
pytorch/sagemakercv/detection/rpn/loss.py	123
pytorch/sagemakercv/detection/roi_heads/__init__.py	1
pytorch/sagemakercv/detection/roi_heads/roi_heads.py	46
pytorch/sagemakercv/detection/roi_heads/box_head/roi_box_predictors.py	69
pytorch/sagemakercv/detection/roi_heads/box_head/__init__.py	1
pytorch/sagemakercv/detection/roi_heads/box_head/inference.py	110
pytorch/sagemakercv/detection/roi_heads/box_head/roi_box_feature_extractors.py	127
pytorch/sagemakercv/detection/roi_heads/box_head/box_head.py	79
pytorch/sagemakercv/detection/roi_heads/box_head/loss.py	306
pytorch/sagemakercv/detection/roi_heads/mask_head/mask_head.py	46
pytorch/sagemakercv/detection/roi_heads/mask_head/__init__.py	1
pytorch/sagemakercv/detection/roi_heads/mask_head/inference.py	127
pytorch/sagemakercv/detection/roi_heads/mask_head/roi_mask_predictors.py	40
pytorch/sagemakercv/detection/roi_heads/mask_head/roi_mask_feature_extractors.py	47
pytorch/sagemakercv/detection/roi_heads/mask_head/loss.py	90
pytorch/sagemakercv/detection/roi_heads/keypoint_head/__init__.py	1
pytorch/sagemakercv/detection/roi_heads/keypoint_head/roi_keypoint_predictors.py	31
pytorch/sagemakercv/detection/roi_heads/keypoint_head/keypoint_head.py	26
pytorch/sagemakercv/detection/roi_heads/keypoint_head/inference.py	82
pytorch/sagemakercv/detection/roi_heads/keypoint_head/roi_keypoint_feature_extractors.py	41
pytorch/sagemakercv/detection/roi_heads/keypoint_head/loss.py	116
pytorch/sagemakercv/inference/__init__.py	1
pytorch/sagemakercv/inference/inference.py	87
pytorch/sagemakercv/inference/tester.py	38
pytorch/sagemakercv/training/trainers.py	17
pytorch/sagemakercv/training/build.py	81
pytorch/sagemakercv/training/__init__.py	4
pytorch/sagemakercv/training/optimizers/fused_novograd.py	118
pytorch/sagemakercv/training/optimizers/__init__.py	2
pytorch/sagemakercv/training/optimizers/schedulers/__init__.py	2
pytorch/sagemakercv/training/optimizers/schedulers/cosine_lr_scheduler.py	56
pytorch/sagemakercv/training/optimizers/schedulers/lr_scheduler.py	46
pytorch/sagemakercv/training/optimizers/mlperf_fused_sgd.py	90
pytorch/sagemakercv/training/optimizers/mlperf_fp16_optimizer.py	158
pytorch/sagemakercv/core/__init__.py	1
pytorch/sagemakercv/core/matcher.py	46
pytorch/sagemakercv/core/poolers.py	72
pytorch/sagemakercv/core/balanced_positive_negative_sampler.py	122
pytorch/sagemakercv/core/utils.py	6
pytorch/sagemakercv/core/structures/image_list.py	65
pytorch/sagemakercv/core/structures/__init__.py	1
pytorch/sagemakercv/core/structures/keypoint.py	147
pytorch/sagemakercv/core/structures/bounding_box.py	216
pytorch/sagemakercv/core/structures/boxlist_ops.py	65
pytorch/sagemakercv/core/structures/segmentation_mask.py	155
pytorch/sagemakercv/core/box_coder.py	52
pytorch/sagemakercv/utils/registry.py	22
pytorch/sagemakercv/utils/metric_logger.py	49
pytorch/sagemakercv/utils/__init__.py	1
pytorch/sagemakercv/utils/mlperf_logger.py	82
pytorch/sagemakercv/utils/loss_utils.py	33
pytorch/sagemakercv/utils/collect_env.py	8
pytorch/sagemakercv/utils/miscellaneous.py	8
pytorch/sagemakercv/utils/fileio/__init__.py	7
pytorch/sagemakercv/utils/fileio/io.py	64
pytorch/sagemakercv/utils/fileio/handlers/json_handler.py	21
pytorch/sagemakercv/utils/fileio/handlers/yaml_handler.py	16
pytorch/sagemakercv/utils/fileio/handlers/__init__.py	5
pytorch/sagemakercv/utils/fileio/handlers/base.py	18
pytorch/sagemakercv/utils/fileio/handlers/pickle_handler.py	17
pytorch/sagemakercv/utils/fileio/parse.py	22
pytorch/sagemakercv/utils/async_evaluator.py	53
pytorch/sagemakercv/utils/imports.py	11
pytorch/sagemakercv/utils/logger.py	19
pytorch/sagemakercv/utils/env.py	20
pytorch/sagemakercv/utils/cv2_util.py	10
pytorch/sagemakercv/utils/dist_utils.py	65
pytorch/sagemakercv/utils/model_zoo.py	29
pytorch/sagemakercv/utils/runner/__init__.py	5
pytorch/sagemakercv/utils/runner/runner.py	158
pytorch/sagemakercv/utils/runner/priority.py	20
pytorch/sagemakercv/utils/runner/builder.py	4
pytorch/sagemakercv/utils/runner/log_buffer.py	39
pytorch/sagemakercv/utils/runner/hooks/coco_evaluation_hook.py	55
pytorch/sagemakercv/utils/runner/hooks/iter_timer.py	26
pytorch/sagemakercv/utils/runner/hooks/__init__.py	8
pytorch/sagemakercv/utils/runner/hooks/amp_optimizer.py	24
pytorch/sagemakercv/utils/runner/hooks/logger/__init__.py	1
pytorch/sagemakercv/utils/runner/hooks/logger/text.py	71
pytorch/sagemakercv/utils/runner/hooks/logger/base.py	37
pytorch/sagemakercv/utils/runner/hooks/hook.py	37
pytorch/sagemakercv/utils/runner/hooks/fp16_optimizer.py	16
pytorch/sagemakercv/utils/runner/hooks/checkpoint.py	48
pytorch/sagemakercv/utils/comm.py	98
pytorch/sagemakercv/utils/model_serialization.py	58
pytorch/sagemakercv/utils/c2_model_loading.py	120
pytorch/sagemakercv/utils/visualize.py	167
pytorch/sagemakercv/utils/checkpoint.py	148
pytorch/sagemakercv/data/datasets/__init__.py	4
pytorch/sagemakercv/data/datasets/voc.py	107
pytorch/sagemakercv/data/datasets/coco.py	337
pytorch/sagemakercv/data/datasets/concat_dataset.py	13
pytorch/sagemakercv/data/datasets/list_dataset.py	17
pytorch/sagemakercv/data/datasets/evaluation/coco/__init__.py	19
pytorch/sagemakercv/data/datasets/evaluation/coco/coco_labels.py	82
pytorch/sagemakercv/data/datasets/evaluation/coco/coco_eval.py	350
pytorch/sagemakercv/data/datasets/evaluation/voc/voc_eval.py	134
pytorch/sagemakercv/data/datasets/evaluation/voc/__init__.py	13
pytorch/sagemakercv/data/datasets/evaluation/__init__.py	16
pytorch/sagemakercv/data/build.py	165
pytorch/sagemakercv/data/__init__.py	1
pytorch/sagemakercv/data/samplers/__init__.py	4
pytorch/sagemakercv/data/samplers/iteration_based_batch_sampler.py	23
pytorch/sagemakercv/data/samplers/distributed.py	44
pytorch/sagemakercv/data/samplers/grouped_batch_sampler.py	57
pytorch/sagemakercv/data/collate_batch.py	19
pytorch/sagemakercv/data/transforms/build.py	32
pytorch/sagemakercv/data/transforms/__init__.py	6
pytorch/sagemakercv/data/transforms/transforms.py	105
pytorch/sagemakercv/layers/pisa_loss.py	142
pytorch/sagemakercv/layers/__init__.py	19
pytorch/sagemakercv/layers/make_layers.py	105
pytorch/sagemakercv/layers/batch_norm.py	16
pytorch/sagemakercv/layers/ce_loss.py	97
pytorch/sagemakercv/layers/roi_pool.py	53
pytorch/sagemakercv/layers/misc.py	74
pytorch/sagemakercv/layers/iou_loss.py	172
pytorch/sagemakercv/layers/nhwc/__init__.py	11
pytorch/sagemakercv/layers/nhwc/batch_norm.py	17
pytorch/sagemakercv/layers/nhwc/transforms.py	31
pytorch/sagemakercv/layers/nhwc/max_pool.py	37
pytorch/sagemakercv/layers/nhwc/conv.py	212
pytorch/sagemakercv/layers/nhwc/UpSampleNearest2d.py	29
pytorch/sagemakercv/layers/nhwc/misc.py	82
pytorch/sagemakercv/layers/nhwc/init.py	35
pytorch/sagemakercv/layers/sigmoid_focal_loss.py	61
pytorch/sagemakercv/layers/roi_align.py	65
pytorch/sagemakercv/layers/nms.py	3
pytorch/sagemakercv/layers/smooth_l1_loss.py	10
pytorch/sagemakercv/layers/_utils.py	29
pytorch/sagemakercv/layers/conv_module.py	41
pytorch/setup.py	38
pytorch/configs/__init__.py	1
pytorch/configs/config.py	170
pytorch/utils.py	47
pytorch/cuda_utils/setup.py	62
pytorch/cuda_utils/smcv_utils/nms_batched.h	15
pytorch/cuda_utils/smcv_utils/rpn_generate_proposals_batched.h	26
pytorch/cuda_utils/smcv_utils/box_iou.h	9
pytorch/cuda_utils/smcv_utils/box_encode.h	9
pytorch/cuda_utils/smcv_utils/match_proposals.h	9
pytorch/cuda_utils/smcv_utils/generate_mask_targets.h	9
pytorch/cuda_utils/smcv_utils/ROIPool.h	39
pytorch/cuda_utils/smcv_utils/anchor_generator.h	11
pytorch/cuda_utils/smcv_utils/cpu/ROIAlign_cpu.cpp	208
pytorch/cuda_utils/smcv_utils/cpu/nms_cpu.cpp	62
pytorch/cuda_utils/smcv_utils/cpu/vision.h	11
pytorch/cuda_utils/smcv_utils/SigmoidFocalLoss.h	36
pytorch/cuda_utils/smcv_utils/ROIAlign.h	41
pytorch/cuda_utils/smcv_utils/vision.cpp	32
pytorch/cuda_utils/smcv_utils/nms.h	21
pytorch/cuda_utils/smcv_utils/cuda/generate_mask_targets.cu	425
pytorch/cuda_utils/smcv_utils/cuda/nms.cu	108
pytorch/cuda_utils/smcv_utils/cuda/nhwc.cpp	17
pytorch/cuda_utils/smcv_utils/cuda/rpn_generate_proposals.cu	204
pytorch/cuda_utils/smcv_utils/cuda/box_iou.cu	79
pytorch/cuda_utils/smcv_utils/cuda/anchor_generator.cu	126
pytorch/cuda_utils/smcv_utils/cuda/box_encode.cu	86
pytorch/cuda_utils/smcv_utils/cuda/rpn_generate_proposals_batched.cu	233
pytorch/cuda_utils/smcv_utils/cuda/vision.h	90
pytorch/cuda_utils/smcv_utils/cuda/nms_batched.cu	166
pytorch/cuda_utils/smcv_utils/cuda/rpn_generate_proposals.h	20
pytorch/cuda_utils/smcv_utils/cuda/match_proposals.cu	162
pytorch/cuda_utils/smcv_utils/cuda/ROIAlign_cuda.cu	509
pytorch/cuda_utils/smcv_utils/cuda/nhwc/Descriptors.cpp	148
pytorch/cuda_utils/smcv_utils/cuda/nhwc/Exceptions.h	66
pytorch/cuda_utils/smcv_utils/cuda/nhwc/Types.h	7
pytorch/cuda_utils/smcv_utils/cuda/nhwc/conv.cpp	859
pytorch/cuda_utils/smcv_utils/cuda/nhwc/Types.cpp	19
pytorch/cuda_utils/smcv_utils/cuda/nhwc/LaunchUtils.h	14
pytorch/cuda_utils/smcv_utils/cuda/nhwc/ParamsHash.h	27
pytorch/cuda_utils/smcv_utils/cuda/nhwc/UpSample.cuh	59
pytorch/cuda_utils/smcv_utils/cuda/nhwc/Descriptors.h	180
pytorch/cuda_utils/smcv_utils/cuda/nhwc/max_pool.cu	112
pytorch/cuda_utils/smcv_utils/cuda/nhwc/UpSampleNearest2d.cu	268
pytorch/cuda_utils/smcv_utils/cuda/nhwc/transforms.cpp	58
pytorch/cuda_utils/smcv_utils/cuda/nhwc.h	56
pytorch/cuda_utils/smcv_utils/cuda/ROIPool_cuda.cu	175
pytorch/cuda_utils/smcv_utils/cuda/SigmoidFocalLoss_cuda.cu	150
tensorflow/train.py	44
tensorflow/sagemakercv/__init__.py	2
tensorflow/sagemakercv/detection/__init__.py	6
tensorflow/sagemakercv/detection/backbones/__init__.py	1
tensorflow/sagemakercv/detection/backbones/resnet.py	521
tensorflow/sagemakercv/detection/detectors/__init__.py	1
tensorflow/sagemakercv/detection/detectors/two_stage_detector.py	52
tensorflow/sagemakercv/detection/necks/__init__.py	1
tensorflow/sagemakercv/detection/necks/fpn.py	81
tensorflow/sagemakercv/detection/dense_heads/__init__.py	2
tensorflow/sagemakercv/detection/dense_heads/anchor_head.py	86
tensorflow/sagemakercv/detection/dense_heads/rpn_head.py	107
tensorflow/sagemakercv/detection/roi_heads/mask_heads/__init__.py	1
tensorflow/sagemakercv/detection/roi_heads/mask_heads/standard_mask_head.py	129
tensorflow/sagemakercv/detection/roi_heads/__init__.py	4
tensorflow/sagemakercv/detection/roi_heads/bbox_heads/standard_bbox_head.py	93
tensorflow/sagemakercv/detection/roi_heads/bbox_heads/__init__.py	1
tensorflow/sagemakercv/detection/roi_heads/cascade_roi_head.py	160
tensorflow/sagemakercv/detection/roi_heads/standard_roi_head.py	174
tensorflow/sagemakercv/detection/builder.py	26
tensorflow/sagemakercv/training/trainers.py	89
tensorflow/sagemakercv/training/__init__.py	4
tensorflow/sagemakercv/training/losses/rpn_losses.py	91
tensorflow/sagemakercv/training/losses/__init__.py	3
tensorflow/sagemakercv/training/losses/losses.py	278
tensorflow/sagemakercv/training/losses/frcnn_losses.py	200
tensorflow/sagemakercv/training/optimizers/__init__.py	3
tensorflow/sagemakercv/training/optimizers/schedulers/__init__.py	2
tensorflow/sagemakercv/training/optimizers/schedulers/schedulers.py	14
tensorflow/sagemakercv/training/optimizers/schedulers/warmup_scheduler.py	37
tensorflow/sagemakercv/training/optimizers/weight_decay_optimizer.py	143
tensorflow/sagemakercv/training/optimizers/novograd.py	175
tensorflow/sagemakercv/training/optimizers/momentum.py	90
tensorflow/sagemakercv/training/builder.py	27
tensorflow/sagemakercv/core/training_ops.py	184
tensorflow/sagemakercv/core/ops.py	20
tensorflow/sagemakercv/core/minibatch_sampler.py	21
tensorflow/sagemakercv/core/__init__.py	13
tensorflow/sagemakercv/core/matcher.py	53
tensorflow/sagemakercv/core/spatial_transform_ops.py	184
tensorflow/sagemakercv/core/argmax_matcher.py	84
tensorflow/sagemakercv/core/preprocessor.py	160
tensorflow/sagemakercv/core/faster_rcnn_box_coder.py	47
tensorflow/sagemakercv/core/postprocess_ops.py	181
tensorflow/sagemakercv/core/preprocess_ops.py	113
tensorflow/sagemakercv/core/region_similarity_calculator.py	38
tensorflow/sagemakercv/core/shape_utils.py	38
tensorflow/sagemakercv/core/box_list.py	58
tensorflow/sagemakercv/core/balanced_positive_negative_sampler.py	118
tensorflow/sagemakercv/core/roi_ops.py	242
tensorflow/sagemakercv/core/box_utils.py	392
tensorflow/sagemakercv/core/box_coder.py	38
tensorflow/sagemakercv/core/builder.py	56
tensorflow/sagemakercv/core/target_assigner.py	93
tensorflow/sagemakercv/core/nms_ops.py	99
tensorflow/sagemakercv/core/anchors.py	126
tensorflow/sagemakercv/utils/registry.py	22
tensorflow/sagemakercv/utils/__init__.py	1
tensorflow/sagemakercv/utils/paths_catalog.py	15
tensorflow/sagemakercv/utils/visualization.py	85
tensorflow/sagemakercv/utils/fileio/__init__.py	7
tensorflow/sagemakercv/utils/fileio/io.py	64
tensorflow/sagemakercv/utils/fileio/handlers/json_handler.py	21
tensorflow/sagemakercv/utils/fileio/handlers/yaml_handler.py	16
tensorflow/sagemakercv/utils/fileio/handlers/__init__.py	5
tensorflow/sagemakercv/utils/fileio/handlers/base.py	18
tensorflow/sagemakercv/utils/fileio/handlers/pickle_handler.py	17
tensorflow/sagemakercv/utils/fileio/parse.py	22
tensorflow/sagemakercv/utils/imports.py	10
tensorflow/sagemakercv/utils/logger.py	9
tensorflow/sagemakercv/utils/metaclasses.py	9
tensorflow/sagemakercv/utils/dist_utils.py	59
tensorflow/sagemakercv/utils/runner/__init__.py	7
tensorflow/sagemakercv/utils/runner/runner.py	181
tensorflow/sagemakercv/utils/runner/priority.py	20
tensorflow/sagemakercv/utils/runner/builder.py	4
tensorflow/sagemakercv/utils/runner/log_buffer.py	37
tensorflow/sagemakercv/utils/runner/hooks/iter_timer.py	27
tensorflow/sagemakercv/utils/runner/hooks/__init__.py	20
tensorflow/sagemakercv/utils/runner/hooks/image_visualizer.py	71
tensorflow/sagemakercv/utils/runner/hooks/logger/__init__.py	2
tensorflow/sagemakercv/utils/runner/hooks/logger/text.py	70
tensorflow/sagemakercv/utils/runner/hooks/logger/tensorboard.py	42
tensorflow/sagemakercv/utils/runner/hooks/logger/base.py	37
tensorflow/sagemakercv/utils/runner/hooks/system_monitor.py	60
tensorflow/sagemakercv/utils/runner/hooks/graph_visualizer.py	24
tensorflow/sagemakercv/utils/runner/hooks/hook.py	37
tensorflow/sagemakercv/utils/runner/hooks/evaluation.py	161
tensorflow/sagemakercv/utils/runner/hooks/profiler.py	17
tensorflow/sagemakercv/utils/runner/hooks/checkpoint.py	50
tensorflow/sagemakercv/utils/logging_formatter.py	181
tensorflow/sagemakercv/data/coco/__init__.py	1
tensorflow/sagemakercv/data/coco/coco_utils.py	124
tensorflow/sagemakercv/data/coco/create_coco_tf_record.py	223
tensorflow/sagemakercv/data/coco/dataloader.py	347
tensorflow/sagemakercv/data/coco/coco_labels.py	185
tensorflow/sagemakercv/data/coco/coco_metric.py	258
tensorflow/sagemakercv/data/coco/tf_example_decoder.py	88
tensorflow/sagemakercv/data/coco/evaluation.py	148
tensorflow/sagemakercv/data/coco/dataloader_utils.py	257
tensorflow/sagemakercv/data/__init__.py	2
tensorflow/sagemakercv/data/builder.py	39
tensorflow/sagemakercv/layers/__init__.py	1
tensorflow/sagemakercv/layers/evo_norm.py	32
tensorflow/sagemakercv/layers/conv_module.py	47
tensorflow/setup.py	24
tensorflow/configs/__init__.py	1
tensorflow/configs/default_config.py	130