├── .dockerignore ├── LICENSE ├── README.md ├── coco_to_waymo.py ├── collect_annotations.py ├── data └── __init__.py ├── detection_results.md ├── detnet ├── .dockerignore ├── configs │ ├── detectron2.cfg │ ├── efficientdet.cfg │ ├── mmdet.cfg │ └── retina.cfg ├── data │ ├── anchor_box_dataset.py │ ├── coco.py │ ├── image_folder.py │ └── metric.py ├── demo.py ├── ensemble.py ├── ensemble_b.py ├── eval.py ├── export.py ├── inference.py ├── nn │ ├── __init__.py │ ├── __main__.py │ ├── basenet │ │ ├── README.md │ │ ├── __init__.py │ │ ├── __main__.py │ │ ├── basic.py │ │ ├── batch_norm.py │ │ ├── darknet.py │ │ ├── deform_conv.py │ │ ├── det_net.py │ │ ├── inplace_abn │ │ │ ├── __init__.py │ │ │ ├── abn.py │ │ │ ├── bn.py │ │ │ ├── functions.py │ │ │ └── src │ │ │ │ ├── common.h │ │ │ │ ├── inplace_abn.cpp │ │ │ │ ├── inplace_abn.h │ │ │ │ ├── inplace_abn_cpu.cpp │ │ │ │ └── inplace_abn_cuda.cu │ │ ├── mobile_net_v2.py │ │ ├── resnet_ibn_a.py │ │ ├── tf_like.py │ │ └── utils.py │ ├── detectron2_det │ │ ├── __init__.py │ │ └── configs │ │ │ ├── Base-RCNN-FPN.yaml │ │ │ ├── COCO-Detection │ │ │ ├── faster_cascade_rcnn_ResNeSt_200_FPN_syncbn_range-scale_1x.yaml │ │ │ ├── faster_rcnn_ResNeSt_101_FPN_syncbn_range-scale_1x.yaml │ │ │ └── faster_rcnn_ResNeSt_50_FPN_syncbn_range-scale_1x.yaml │ │ │ └── COCO-InstanceSegmentation │ │ │ ├── mask_cascade_rcnn_ResNeSt_101_FPN_syncBN_1x.yaml │ │ │ └── mask_cascade_rcnn_ResNeSt_50_FPN_syncBN_1x.yaml │ ├── loss │ │ ├── cross_entropy.py │ │ ├── focal.py │ │ ├── lovasz.py │ │ ├── mask.py │ │ ├── multi_box.py │ │ ├── smooth_l1.py │ │ └── ssd.py │ ├── mm_det │ │ ├── __init__.py │ │ └── configs │ │ │ ├── cascade_mask_rcnn_r50_fpn.py │ │ │ ├── faster_rcnn_r50_fpn.py │ │ │ ├── htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn.py │ │ │ └── mask_rcnn_r50_fpn.py │ ├── modules │ │ ├── activation.py │ │ ├── bifpn.py │ │ ├── bounding_box.py │ │ ├── danet.py │ │ ├── decoder.py │ │ ├── det_head.py │ │ ├── detection.py │ │ ├── l2norm.py │ │ ├── pooling.py │ │ ├── pose_resnet.py │ │ ├── prior_box.py │ │ ├── pyramid_feature.py │ │ └── receptive_field_block.py │ ├── point_det.py │ ├── ssd.py │ ├── torchvision_det.py │ └── tta.py ├── requirements.txt ├── swa.py ├── to_onnx.py ├── train.py ├── trainer │ ├── README.md │ ├── __init__.py │ ├── compute_mean_and_std.py │ ├── data │ │ ├── __init__.py │ │ ├── dataset.py │ │ └── image_folder.py │ ├── doc │ │ ├── pytorch_train_process.png │ │ └── pytorch_train_process.tex │ ├── example │ │ └── mnist.py │ ├── launch.sh │ ├── loss │ │ └── __init__.py │ ├── mixup.py │ ├── optim │ │ ├── __init__.py │ │ ├── adabound.py │ │ ├── lookahead.py │ │ ├── lr_scheduler.py │ │ ├── novograd.py │ │ └── radam.py │ ├── predictions.py │ ├── swa.py │ ├── test.py │ ├── train.py │ ├── transforms │ │ ├── __init__.py │ │ ├── auto_aug.py │ │ ├── common.py │ │ └── vision.py │ └── utils │ │ ├── __init__.py │ │ ├── collect_env.py │ │ └── rle.py ├── transforms │ ├── test_transforms.py │ └── train_transforms.py └── utils │ ├── box_utils.py │ ├── rbox_utils.py │ └── visualization.py ├── doc └── waymo_2d_tracking_dainamite.pdf ├── environment.yml ├── generate_prediction_for_metrics.py ├── inference.py ├── logs ├── 12442 │ ├── events.out.tfevents.1589568144.lambda-server │ └── job.log ├── 12620 │ ├── events.out.tfevents.1590426656.lambda-server │ └── job.log └── 12650 │ ├── events.out.tfevents.1590580726.lambda-server │ └── job.log ├── tracking ├── sort │ ├── __init__.py │ ├── sort.py │ └── tracker_sort.py ├── track.py ├── utils.py └── visualize.py ├── train.py └── waymo_to_coco.py /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/.dockerignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/README.md -------------------------------------------------------------------------------- /coco_to_waymo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/coco_to_waymo.py -------------------------------------------------------------------------------- /collect_annotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/collect_annotations.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/data/__init__.py -------------------------------------------------------------------------------- /detection_results.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detection_results.md -------------------------------------------------------------------------------- /detnet/.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/.dockerignore -------------------------------------------------------------------------------- /detnet/configs/detectron2.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/configs/detectron2.cfg -------------------------------------------------------------------------------- /detnet/configs/efficientdet.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/configs/efficientdet.cfg -------------------------------------------------------------------------------- /detnet/configs/mmdet.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/configs/mmdet.cfg -------------------------------------------------------------------------------- /detnet/configs/retina.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/configs/retina.cfg -------------------------------------------------------------------------------- /detnet/data/anchor_box_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/data/anchor_box_dataset.py -------------------------------------------------------------------------------- /detnet/data/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/data/coco.py -------------------------------------------------------------------------------- /detnet/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/data/image_folder.py -------------------------------------------------------------------------------- /detnet/data/metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/data/metric.py -------------------------------------------------------------------------------- /detnet/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/demo.py -------------------------------------------------------------------------------- /detnet/ensemble.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/ensemble.py -------------------------------------------------------------------------------- /detnet/ensemble_b.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/ensemble_b.py -------------------------------------------------------------------------------- /detnet/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/eval.py -------------------------------------------------------------------------------- /detnet/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/export.py -------------------------------------------------------------------------------- /detnet/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/inference.py -------------------------------------------------------------------------------- /detnet/nn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/__init__.py -------------------------------------------------------------------------------- /detnet/nn/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/__main__.py -------------------------------------------------------------------------------- /detnet/nn/basenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/README.md -------------------------------------------------------------------------------- /detnet/nn/basenet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/__init__.py -------------------------------------------------------------------------------- /detnet/nn/basenet/__main__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/__main__.py -------------------------------------------------------------------------------- /detnet/nn/basenet/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/basic.py -------------------------------------------------------------------------------- /detnet/nn/basenet/batch_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/batch_norm.py -------------------------------------------------------------------------------- /detnet/nn/basenet/darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/darknet.py -------------------------------------------------------------------------------- /detnet/nn/basenet/deform_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/deform_conv.py -------------------------------------------------------------------------------- /detnet/nn/basenet/det_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/det_net.py -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/__init__.py -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/abn.py -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/bn.py -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/functions.py -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/src/common.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/src/common.h -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/src/inplace_abn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/src/inplace_abn.cpp -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/src/inplace_abn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/src/inplace_abn.h -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/src/inplace_abn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/src/inplace_abn_cpu.cpp -------------------------------------------------------------------------------- /detnet/nn/basenet/inplace_abn/src/inplace_abn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/inplace_abn/src/inplace_abn_cuda.cu -------------------------------------------------------------------------------- /detnet/nn/basenet/mobile_net_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/mobile_net_v2.py -------------------------------------------------------------------------------- /detnet/nn/basenet/resnet_ibn_a.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/resnet_ibn_a.py -------------------------------------------------------------------------------- /detnet/nn/basenet/tf_like.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/tf_like.py -------------------------------------------------------------------------------- /detnet/nn/basenet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/basenet/utils.py -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/__init__.py -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/Base-RCNN-FPN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/Base-RCNN-FPN.yaml -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/COCO-Detection/faster_cascade_rcnn_ResNeSt_200_FPN_syncbn_range-scale_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/COCO-Detection/faster_cascade_rcnn_ResNeSt_200_FPN_syncbn_range-scale_1x.yaml -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/COCO-Detection/faster_rcnn_ResNeSt_101_FPN_syncbn_range-scale_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/COCO-Detection/faster_rcnn_ResNeSt_101_FPN_syncbn_range-scale_1x.yaml -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/COCO-Detection/faster_rcnn_ResNeSt_50_FPN_syncbn_range-scale_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/COCO-Detection/faster_rcnn_ResNeSt_50_FPN_syncbn_range-scale_1x.yaml -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/COCO-InstanceSegmentation/mask_cascade_rcnn_ResNeSt_101_FPN_syncBN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/COCO-InstanceSegmentation/mask_cascade_rcnn_ResNeSt_101_FPN_syncBN_1x.yaml -------------------------------------------------------------------------------- /detnet/nn/detectron2_det/configs/COCO-InstanceSegmentation/mask_cascade_rcnn_ResNeSt_50_FPN_syncBN_1x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/detectron2_det/configs/COCO-InstanceSegmentation/mask_cascade_rcnn_ResNeSt_50_FPN_syncBN_1x.yaml -------------------------------------------------------------------------------- /detnet/nn/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/cross_entropy.py -------------------------------------------------------------------------------- /detnet/nn/loss/focal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/focal.py -------------------------------------------------------------------------------- /detnet/nn/loss/lovasz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/lovasz.py -------------------------------------------------------------------------------- /detnet/nn/loss/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/mask.py -------------------------------------------------------------------------------- /detnet/nn/loss/multi_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/multi_box.py -------------------------------------------------------------------------------- /detnet/nn/loss/smooth_l1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/smooth_l1.py -------------------------------------------------------------------------------- /detnet/nn/loss/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/loss/ssd.py -------------------------------------------------------------------------------- /detnet/nn/mm_det/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/mm_det/__init__.py -------------------------------------------------------------------------------- /detnet/nn/mm_det/configs/cascade_mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/mm_det/configs/cascade_mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detnet/nn/mm_det/configs/faster_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/mm_det/configs/faster_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detnet/nn/mm_det/configs/htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/mm_det/configs/htc_dconv_c3-c5_mstrain_400_1400_x101_64x4d_fpn.py -------------------------------------------------------------------------------- /detnet/nn/mm_det/configs/mask_rcnn_r50_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/mm_det/configs/mask_rcnn_r50_fpn.py -------------------------------------------------------------------------------- /detnet/nn/modules/activation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/activation.py -------------------------------------------------------------------------------- /detnet/nn/modules/bifpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/bifpn.py -------------------------------------------------------------------------------- /detnet/nn/modules/bounding_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/bounding_box.py -------------------------------------------------------------------------------- /detnet/nn/modules/danet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/danet.py -------------------------------------------------------------------------------- /detnet/nn/modules/decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/decoder.py -------------------------------------------------------------------------------- /detnet/nn/modules/det_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/det_head.py -------------------------------------------------------------------------------- /detnet/nn/modules/detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/detection.py -------------------------------------------------------------------------------- /detnet/nn/modules/l2norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/l2norm.py -------------------------------------------------------------------------------- /detnet/nn/modules/pooling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/pooling.py -------------------------------------------------------------------------------- /detnet/nn/modules/pose_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/pose_resnet.py -------------------------------------------------------------------------------- /detnet/nn/modules/prior_box.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/prior_box.py -------------------------------------------------------------------------------- /detnet/nn/modules/pyramid_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/pyramid_feature.py -------------------------------------------------------------------------------- /detnet/nn/modules/receptive_field_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/modules/receptive_field_block.py -------------------------------------------------------------------------------- /detnet/nn/point_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/point_det.py -------------------------------------------------------------------------------- /detnet/nn/ssd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/ssd.py -------------------------------------------------------------------------------- /detnet/nn/torchvision_det.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/torchvision_det.py -------------------------------------------------------------------------------- /detnet/nn/tta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/nn/tta.py -------------------------------------------------------------------------------- /detnet/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/requirements.txt -------------------------------------------------------------------------------- /detnet/swa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/swa.py -------------------------------------------------------------------------------- /detnet/to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/to_onnx.py -------------------------------------------------------------------------------- /detnet/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/train.py -------------------------------------------------------------------------------- /detnet/trainer/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/README.md -------------------------------------------------------------------------------- /detnet/trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/__init__.py -------------------------------------------------------------------------------- /detnet/trainer/compute_mean_and_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/compute_mean_and_std.py -------------------------------------------------------------------------------- /detnet/trainer/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/data/__init__.py -------------------------------------------------------------------------------- /detnet/trainer/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/data/dataset.py -------------------------------------------------------------------------------- /detnet/trainer/data/image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/data/image_folder.py -------------------------------------------------------------------------------- /detnet/trainer/doc/pytorch_train_process.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/doc/pytorch_train_process.png -------------------------------------------------------------------------------- /detnet/trainer/doc/pytorch_train_process.tex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/doc/pytorch_train_process.tex -------------------------------------------------------------------------------- /detnet/trainer/example/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/example/mnist.py -------------------------------------------------------------------------------- /detnet/trainer/launch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/launch.sh -------------------------------------------------------------------------------- /detnet/trainer/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/loss/__init__.py -------------------------------------------------------------------------------- /detnet/trainer/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/mixup.py -------------------------------------------------------------------------------- /detnet/trainer/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/__init__.py -------------------------------------------------------------------------------- /detnet/trainer/optim/adabound.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/adabound.py -------------------------------------------------------------------------------- /detnet/trainer/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/lookahead.py -------------------------------------------------------------------------------- /detnet/trainer/optim/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/lr_scheduler.py -------------------------------------------------------------------------------- /detnet/trainer/optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/novograd.py -------------------------------------------------------------------------------- /detnet/trainer/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/optim/radam.py -------------------------------------------------------------------------------- /detnet/trainer/predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/predictions.py -------------------------------------------------------------------------------- /detnet/trainer/swa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/swa.py -------------------------------------------------------------------------------- /detnet/trainer/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/test.py -------------------------------------------------------------------------------- /detnet/trainer/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/train.py -------------------------------------------------------------------------------- /detnet/trainer/transforms/__init__.py: -------------------------------------------------------------------------------- 1 | from .common import * -------------------------------------------------------------------------------- /detnet/trainer/transforms/auto_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/transforms/auto_aug.py -------------------------------------------------------------------------------- /detnet/trainer/transforms/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/transforms/common.py -------------------------------------------------------------------------------- /detnet/trainer/transforms/vision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/transforms/vision.py -------------------------------------------------------------------------------- /detnet/trainer/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/utils/__init__.py -------------------------------------------------------------------------------- /detnet/trainer/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/utils/collect_env.py -------------------------------------------------------------------------------- /detnet/trainer/utils/rle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/trainer/utils/rle.py -------------------------------------------------------------------------------- /detnet/transforms/test_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/transforms/test_transforms.py -------------------------------------------------------------------------------- /detnet/transforms/train_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/transforms/train_transforms.py -------------------------------------------------------------------------------- /detnet/utils/box_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/utils/box_utils.py -------------------------------------------------------------------------------- /detnet/utils/rbox_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/utils/rbox_utils.py -------------------------------------------------------------------------------- /detnet/utils/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/detnet/utils/visualization.py -------------------------------------------------------------------------------- /doc/waymo_2d_tracking_dainamite.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/doc/waymo_2d_tracking_dainamite.pdf -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/environment.yml -------------------------------------------------------------------------------- /generate_prediction_for_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/generate_prediction_for_metrics.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/inference.py -------------------------------------------------------------------------------- /logs/12442/events.out.tfevents.1589568144.lambda-server: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12442/events.out.tfevents.1589568144.lambda-server -------------------------------------------------------------------------------- /logs/12442/job.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12442/job.log -------------------------------------------------------------------------------- /logs/12620/events.out.tfevents.1590426656.lambda-server: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12620/events.out.tfevents.1590426656.lambda-server -------------------------------------------------------------------------------- /logs/12620/job.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12620/job.log -------------------------------------------------------------------------------- /logs/12650/events.out.tfevents.1590580726.lambda-server: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12650/events.out.tfevents.1590580726.lambda-server -------------------------------------------------------------------------------- /logs/12650/job.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/logs/12650/job.log -------------------------------------------------------------------------------- /tracking/sort/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tracking/sort/sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/tracking/sort/sort.py -------------------------------------------------------------------------------- /tracking/sort/tracker_sort.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/tracking/sort/tracker_sort.py -------------------------------------------------------------------------------- /tracking/track.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/tracking/track.py -------------------------------------------------------------------------------- /tracking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/tracking/utils.py -------------------------------------------------------------------------------- /tracking/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/tracking/visualize.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/train.py -------------------------------------------------------------------------------- /waymo_to_coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xuyuan/waymo_2d_tracking/HEAD/waymo_to_coco.py --------------------------------------------------------------------------------