├── .gitignore ├── README.md ├── cfgs ├── __init__.py ├── config.py ├── config_crowdai.yaml ├── config_detrac.yaml ├── config_kitti.py ├── config_kitti.yaml ├── config_kitti_demo.yaml ├── config_v2.py ├── config_voc.py ├── config_voc.yaml └── exps │ ├── crowdai │ └── crowdai_baseline.yaml │ ├── detrac │ ├── detrac_baseline.yaml │ ├── detrac_baseline_2.yaml │ └── detrac_flow_center_w01.yaml │ ├── kitti │ ├── kitti_baseline.yaml │ ├── kitti_baseline_v3.yaml │ ├── kitti_baseline_v3_yf.yaml │ ├── kitti_joint_flow.yaml │ ├── kitti_joint_flow_low_lr.yaml │ ├── kitti_new_1.yaml │ ├── kitti_new_2.yaml │ ├── kitti_new_2_flow_center_ft.yaml │ ├── kitti_new_2_flow_center_ft_bbox_merge.yaml │ ├── kitti_new_2_flow_center_ft_cls.yaml │ ├── kitti_new_2_flow_center_ft_cls_bbox_merge.yaml │ ├── kitti_new_2_flow_center_ft_cls_flownet2.yaml │ ├── kitti_new_2_flow_center_ft_cls_merge.yaml │ ├── kitti_new_2_flow_center_ft_cls_merge_2x.yaml │ ├── kitti_new_2_flow_center_ft_cls_merge_ped.yaml │ ├── kitti_new_2_flow_center_ft_flownet2.yaml │ ├── kitti_new_2_flow_center_ft_flownet2_joint.yaml │ ├── kitti_new_2_flow_center_ft_half.yaml │ ├── kitti_new_2_flow_center_ft_iou_merge.yaml │ ├── kitti_new_2_flow_center_joint_ft.yaml │ ├── kitti_new_2_flow_center_joint_ft_low_lr.yaml │ ├── kitti_new_2_flow_center_joint_ft_w10.yaml │ ├── kitti_new_2_flow_dis.yaml │ ├── kitti_new_2_flow_ft.yaml │ ├── kitti_new_2_flow_ft_cls_bbox_merge.yaml │ ├── kitti_new_2_flow_ft_cls_merge.yaml │ ├── kitti_new_2_flow_ft_std_2_5.yaml │ ├── kitti_new_2_flow_ft_std_2_5_merge.yaml │ └── kitti_new_2_flow_spy.yaml │ ├── kitti_ft_exp3.py │ ├── voc0712 │ ├── voc0712_anchor.yaml │ ├── voc0712_baseline.yaml │ ├── voc0712_baseline_v3.yaml │ ├── voc0712_baseline_v3_rand.yaml │ ├── voc0712_box_mask_0.yaml │ ├── voc0712_low_lr.yaml │ ├── voc0712_mask.yaml │ ├── voc0712_multiple_anchors.yaml │ ├── voc0712_new.py │ ├── voc0712_obj_scale.yaml │ ├── voc0712_obj_scale_ft.yaml │ ├── voc0712_one_anchor.yaml │ ├── voc0712_overfit.yaml │ ├── voc0712_pred_raw.yaml │ ├── voc0712_template.yaml │ └── voc0712_trainval_ft_debug2.yaml │ └── voc0712_new_2.py ├── darknet.py ├── darknet_training_v3.py ├── darknet_v3.py ├── datasets ├── DataLoaderIterX.py ├── DataLoaderX.py ├── DetectionDataset.py ├── DetectionDatasetHelper.py ├── ImageFileDataset.py ├── ImageFileDataset_v2.py ├── KittiDataset.py ├── __init__.py ├── imdb.py ├── pascal_voc.py └── voc_eval.py ├── demo ├── demo_images_list.txt ├── detection_0030.jpg └── images │ ├── 000040.jpg │ ├── 000041.jpg │ ├── 000042.jpg │ ├── 000043.jpg │ ├── 000044.jpg │ ├── 000045.jpg │ ├── 000046.jpg │ ├── 000047.jpg │ ├── 000048.jpg │ ├── 000049.jpg │ ├── 000050.jpg │ ├── 000051.jpg │ ├── 000052.jpg │ ├── 000053.jpg │ ├── 000054.jpg │ ├── 000055.jpg │ ├── 000056.jpg │ ├── 000057.jpg │ ├── 000058.jpg │ ├── 000059.jpg │ ├── 000060.jpg │ ├── 000061.jpg │ ├── 000062.jpg │ ├── 000063.jpg │ ├── 000064.jpg │ ├── 000065.jpg │ ├── 000066.jpg │ ├── 000067.jpg │ ├── 000068.jpg │ ├── 000069.jpg │ ├── 000070.jpg │ ├── 000071.jpg │ ├── 000072.jpg │ ├── 000073.jpg │ ├── 000074.jpg │ ├── 000075.jpg │ ├── 000076.jpg │ ├── 000077.jpg │ ├── 000078.jpg │ └── 000079.jpg ├── flow ├── detection_analyzer.py ├── flow_generator.py ├── flow_util.py ├── flow_vis.py ├── flow_warper.py ├── gen_flow_images.py ├── gen_flow_images_cloudai.py ├── gen_flow_images_detrac.py ├── gen_val_from_all.sh ├── gen_warp_images_by_flow.py ├── img_diff.py ├── img_pairs.txt ├── kitti_train_labels.txt ├── kitti_val_images_warp.txt ├── plot_util.py ├── run_flow.sh ├── run_of.sh ├── shift_gt_by_flow.py ├── vis.jpg └── yolo_flow.py ├── layers ├── __init__.py ├── reorg │ ├── __init__.py │ ├── _ext │ │ ├── __init__.py │ │ └── reorg_layer │ │ │ └── __init__.py │ ├── build.py │ ├── reorg_layer.py │ └── src │ │ ├── reorg_cpu.c │ │ ├── reorg_cpu.h │ │ ├── reorg_cuda.c │ │ ├── reorg_cuda.h │ │ ├── reorg_cuda_kernel.cu │ │ └── reorg_cuda_kernel.h └── roi_pooling │ ├── __init__.py │ ├── _ext │ ├── __init__.py │ └── roi_pooling │ │ └── __init__.py │ ├── build.py │ ├── roi_pool.py │ ├── roi_pool_py.py │ └── src │ ├── cuda │ ├── roi_pooling_kernel.cu │ └── roi_pooling_kernel.h │ ├── roi_pooling.c │ ├── roi_pooling.h │ ├── roi_pooling_cuda.c │ └── roi_pooling_cuda.h ├── make.sh ├── misc ├── kitti_detect.py ├── validate_dataset.py ├── vis.jpg ├── visualize_gt.py ├── voc_ap.py ├── voc_ap_v2.py ├── voc_data.py └── yolo_video_test.py ├── train ├── train_dataset_v3.py ├── train_util_v2.py └── yellowfin.py ├── train_data ├── gen_dashcam_train_data.py ├── gen_gtav_train_data.py ├── gen_kitti_det_train_data.py ├── gen_kitti_train_data.py └── gen_voc_train_data.py ├── utils ├── __init__.py ├── barrier.py ├── bbox.c ├── bbox.pyx ├── build.py ├── im_transform.py ├── network.py ├── nms │ ├── .gitignore │ ├── __init__.py │ ├── cpu_nms.pyx │ ├── gpu_nms.hpp │ ├── gpu_nms.pyx │ ├── nms_kernel.cu │ └── py_cpu_nms.py ├── nms_wrapper.py ├── plot_loss.py ├── pycocotools │ ├── UPSTREAM_REV │ ├── __init__.py │ ├── _mask.c │ ├── _mask.pyx │ ├── coco.py │ ├── cocoeval.py │ ├── license.txt │ ├── mask.py │ ├── maskApi.c │ └── maskApi.h ├── timer.py ├── vis_util.py ├── yolo.c ├── yolo.py ├── yolo.pyx └── yolo_v2.py └── yolo_detect.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cfgs/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config.py -------------------------------------------------------------------------------- /cfgs/config_crowdai.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_crowdai.yaml -------------------------------------------------------------------------------- /cfgs/config_detrac.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_detrac.yaml -------------------------------------------------------------------------------- /cfgs/config_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_kitti.py -------------------------------------------------------------------------------- /cfgs/config_kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_kitti.yaml -------------------------------------------------------------------------------- /cfgs/config_kitti_demo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_kitti_demo.yaml -------------------------------------------------------------------------------- /cfgs/config_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_v2.py -------------------------------------------------------------------------------- /cfgs/config_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_voc.py -------------------------------------------------------------------------------- /cfgs/config_voc.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/config_voc.yaml -------------------------------------------------------------------------------- /cfgs/exps/crowdai/crowdai_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/crowdai/crowdai_baseline.yaml -------------------------------------------------------------------------------- /cfgs/exps/detrac/detrac_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/detrac/detrac_baseline.yaml -------------------------------------------------------------------------------- /cfgs/exps/detrac/detrac_baseline_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/detrac/detrac_baseline_2.yaml -------------------------------------------------------------------------------- /cfgs/exps/detrac/detrac_flow_center_w01.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/detrac/detrac_flow_center_w01.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_baseline.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_baseline_v3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_baseline_v3.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_baseline_v3_yf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_baseline_v3_yf.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_joint_flow.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_joint_flow.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_joint_flow_low_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_joint_flow_low_lr.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_1.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_bbox_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_bbox_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_bbox_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_bbox_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_flownet2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_flownet2.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge_2x.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge_2x.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge_ped.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_cls_merge_ped.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_flownet2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_flownet2.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_flownet2_joint.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_flownet2_joint.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_half.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_half.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_ft_iou_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_ft_iou_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft_low_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft_low_lr.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft_w10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_center_joint_ft_w10.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_dis.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_dis.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_ft.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_ft_cls_bbox_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_ft_cls_bbox_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_ft_cls_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_ft_cls_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_ft_std_2_5.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_ft_std_2_5.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_ft_std_2_5_merge.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_ft_std_2_5_merge.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti/kitti_new_2_flow_spy.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti/kitti_new_2_flow_spy.yaml -------------------------------------------------------------------------------- /cfgs/exps/kitti_ft_exp3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/kitti_ft_exp3.py -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_anchor.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_baseline.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_baseline_v3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_baseline_v3.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_baseline_v3_rand.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_baseline_v3_rand.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_box_mask_0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_box_mask_0.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_low_lr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_low_lr.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_mask.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_mask.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_multiple_anchors.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_multiple_anchors.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_new.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_new.py -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_obj_scale.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_obj_scale.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_obj_scale_ft.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_obj_scale_ft.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_one_anchor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_one_anchor.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_overfit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_overfit.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_pred_raw.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_pred_raw.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_template.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_template.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712/voc0712_trainval_ft_debug2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712/voc0712_trainval_ft_debug2.yaml -------------------------------------------------------------------------------- /cfgs/exps/voc0712_new_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/cfgs/exps/voc0712_new_2.py -------------------------------------------------------------------------------- /darknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/darknet.py -------------------------------------------------------------------------------- /darknet_training_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/darknet_training_v3.py -------------------------------------------------------------------------------- /darknet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/darknet_v3.py -------------------------------------------------------------------------------- /datasets/DataLoaderIterX.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/DataLoaderIterX.py -------------------------------------------------------------------------------- /datasets/DataLoaderX.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/DataLoaderX.py -------------------------------------------------------------------------------- /datasets/DetectionDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/DetectionDataset.py -------------------------------------------------------------------------------- /datasets/DetectionDatasetHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/DetectionDatasetHelper.py -------------------------------------------------------------------------------- /datasets/ImageFileDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/ImageFileDataset.py -------------------------------------------------------------------------------- /datasets/ImageFileDataset_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/ImageFileDataset_v2.py -------------------------------------------------------------------------------- /datasets/KittiDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/KittiDataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/imdb.py -------------------------------------------------------------------------------- /datasets/pascal_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/pascal_voc.py -------------------------------------------------------------------------------- /datasets/voc_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/datasets/voc_eval.py -------------------------------------------------------------------------------- /demo/demo_images_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/demo_images_list.txt -------------------------------------------------------------------------------- /demo/detection_0030.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/detection_0030.jpg -------------------------------------------------------------------------------- /demo/images/000040.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000040.jpg -------------------------------------------------------------------------------- /demo/images/000041.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000041.jpg -------------------------------------------------------------------------------- /demo/images/000042.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000042.jpg -------------------------------------------------------------------------------- /demo/images/000043.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000043.jpg -------------------------------------------------------------------------------- /demo/images/000044.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000044.jpg -------------------------------------------------------------------------------- /demo/images/000045.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000045.jpg -------------------------------------------------------------------------------- /demo/images/000046.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000046.jpg -------------------------------------------------------------------------------- /demo/images/000047.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000047.jpg -------------------------------------------------------------------------------- /demo/images/000048.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000048.jpg -------------------------------------------------------------------------------- /demo/images/000049.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000049.jpg -------------------------------------------------------------------------------- /demo/images/000050.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000050.jpg -------------------------------------------------------------------------------- /demo/images/000051.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000051.jpg -------------------------------------------------------------------------------- /demo/images/000052.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000052.jpg -------------------------------------------------------------------------------- /demo/images/000053.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000053.jpg -------------------------------------------------------------------------------- /demo/images/000054.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000054.jpg -------------------------------------------------------------------------------- /demo/images/000055.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000055.jpg -------------------------------------------------------------------------------- /demo/images/000056.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000056.jpg -------------------------------------------------------------------------------- /demo/images/000057.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000057.jpg -------------------------------------------------------------------------------- /demo/images/000058.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000058.jpg -------------------------------------------------------------------------------- /demo/images/000059.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000059.jpg -------------------------------------------------------------------------------- /demo/images/000060.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000060.jpg -------------------------------------------------------------------------------- /demo/images/000061.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000061.jpg -------------------------------------------------------------------------------- /demo/images/000062.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000062.jpg -------------------------------------------------------------------------------- /demo/images/000063.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000063.jpg -------------------------------------------------------------------------------- /demo/images/000064.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000064.jpg -------------------------------------------------------------------------------- /demo/images/000065.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000065.jpg -------------------------------------------------------------------------------- /demo/images/000066.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000066.jpg -------------------------------------------------------------------------------- /demo/images/000067.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000067.jpg -------------------------------------------------------------------------------- /demo/images/000068.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000068.jpg -------------------------------------------------------------------------------- /demo/images/000069.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000069.jpg -------------------------------------------------------------------------------- /demo/images/000070.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000070.jpg -------------------------------------------------------------------------------- /demo/images/000071.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000071.jpg -------------------------------------------------------------------------------- /demo/images/000072.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000072.jpg -------------------------------------------------------------------------------- /demo/images/000073.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000073.jpg -------------------------------------------------------------------------------- /demo/images/000074.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000074.jpg -------------------------------------------------------------------------------- /demo/images/000075.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000075.jpg -------------------------------------------------------------------------------- /demo/images/000076.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000076.jpg -------------------------------------------------------------------------------- /demo/images/000077.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000077.jpg -------------------------------------------------------------------------------- /demo/images/000078.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000078.jpg -------------------------------------------------------------------------------- /demo/images/000079.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/demo/images/000079.jpg -------------------------------------------------------------------------------- /flow/detection_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/detection_analyzer.py -------------------------------------------------------------------------------- /flow/flow_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/flow_generator.py -------------------------------------------------------------------------------- /flow/flow_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/flow_util.py -------------------------------------------------------------------------------- /flow/flow_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/flow_vis.py -------------------------------------------------------------------------------- /flow/flow_warper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/flow_warper.py -------------------------------------------------------------------------------- /flow/gen_flow_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/gen_flow_images.py -------------------------------------------------------------------------------- /flow/gen_flow_images_cloudai.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/gen_flow_images_cloudai.py -------------------------------------------------------------------------------- /flow/gen_flow_images_detrac.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/gen_flow_images_detrac.py -------------------------------------------------------------------------------- /flow/gen_val_from_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/gen_val_from_all.sh -------------------------------------------------------------------------------- /flow/gen_warp_images_by_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/gen_warp_images_by_flow.py -------------------------------------------------------------------------------- /flow/img_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/img_diff.py -------------------------------------------------------------------------------- /flow/img_pairs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/img_pairs.txt -------------------------------------------------------------------------------- /flow/kitti_train_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/kitti_train_labels.txt -------------------------------------------------------------------------------- /flow/kitti_val_images_warp.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/kitti_val_images_warp.txt -------------------------------------------------------------------------------- /flow/plot_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/plot_util.py -------------------------------------------------------------------------------- /flow/run_flow.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/run_flow.sh -------------------------------------------------------------------------------- /flow/run_of.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/run_of.sh -------------------------------------------------------------------------------- /flow/shift_gt_by_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/shift_gt_by_flow.py -------------------------------------------------------------------------------- /flow/vis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/vis.jpg -------------------------------------------------------------------------------- /flow/yolo_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/flow/yolo_flow.py -------------------------------------------------------------------------------- /layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/reorg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/reorg/_ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/reorg/_ext/reorg_layer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/_ext/reorg_layer/__init__.py -------------------------------------------------------------------------------- /layers/reorg/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/build.py -------------------------------------------------------------------------------- /layers/reorg/reorg_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/reorg_layer.py -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cpu.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cpu.c -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cpu.h -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cuda.c -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cuda.h -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cuda_kernel.cu -------------------------------------------------------------------------------- /layers/reorg/src/reorg_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/reorg/src/reorg_cuda_kernel.h -------------------------------------------------------------------------------- /layers/roi_pooling/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/roi_pooling/_ext/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /layers/roi_pooling/_ext/roi_pooling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/_ext/roi_pooling/__init__.py -------------------------------------------------------------------------------- /layers/roi_pooling/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/build.py -------------------------------------------------------------------------------- /layers/roi_pooling/roi_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/roi_pool.py -------------------------------------------------------------------------------- /layers/roi_pooling/roi_pool_py.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/roi_pool_py.py -------------------------------------------------------------------------------- /layers/roi_pooling/src/cuda/roi_pooling_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/cuda/roi_pooling_kernel.cu -------------------------------------------------------------------------------- /layers/roi_pooling/src/cuda/roi_pooling_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/cuda/roi_pooling_kernel.h -------------------------------------------------------------------------------- /layers/roi_pooling/src/roi_pooling.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/roi_pooling.c -------------------------------------------------------------------------------- /layers/roi_pooling/src/roi_pooling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/roi_pooling.h -------------------------------------------------------------------------------- /layers/roi_pooling/src/roi_pooling_cuda.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/roi_pooling_cuda.c -------------------------------------------------------------------------------- /layers/roi_pooling/src/roi_pooling_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/layers/roi_pooling/src/roi_pooling_cuda.h -------------------------------------------------------------------------------- /make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/make.sh -------------------------------------------------------------------------------- /misc/kitti_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/kitti_detect.py -------------------------------------------------------------------------------- /misc/validate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/validate_dataset.py -------------------------------------------------------------------------------- /misc/vis.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/vis.jpg -------------------------------------------------------------------------------- /misc/visualize_gt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/visualize_gt.py -------------------------------------------------------------------------------- /misc/voc_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/voc_ap.py -------------------------------------------------------------------------------- /misc/voc_ap_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/voc_ap_v2.py -------------------------------------------------------------------------------- /misc/voc_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/voc_data.py -------------------------------------------------------------------------------- /misc/yolo_video_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/misc/yolo_video_test.py -------------------------------------------------------------------------------- /train/train_dataset_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train/train_dataset_v3.py -------------------------------------------------------------------------------- /train/train_util_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train/train_util_v2.py -------------------------------------------------------------------------------- /train/yellowfin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train/yellowfin.py -------------------------------------------------------------------------------- /train_data/gen_dashcam_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train_data/gen_dashcam_train_data.py -------------------------------------------------------------------------------- /train_data/gen_gtav_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train_data/gen_gtav_train_data.py -------------------------------------------------------------------------------- /train_data/gen_kitti_det_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train_data/gen_kitti_det_train_data.py -------------------------------------------------------------------------------- /train_data/gen_kitti_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train_data/gen_kitti_train_data.py -------------------------------------------------------------------------------- /train_data/gen_voc_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/train_data/gen_voc_train_data.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/barrier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/barrier.py -------------------------------------------------------------------------------- /utils/bbox.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/bbox.c -------------------------------------------------------------------------------- /utils/bbox.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/bbox.pyx -------------------------------------------------------------------------------- /utils/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/build.py -------------------------------------------------------------------------------- /utils/im_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/im_transform.py -------------------------------------------------------------------------------- /utils/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/network.py -------------------------------------------------------------------------------- /utils/nms/.gitignore: -------------------------------------------------------------------------------- 1 | *.c 2 | *.cpp 3 | *.so 4 | -------------------------------------------------------------------------------- /utils/nms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/nms/cpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms/cpu_nms.pyx -------------------------------------------------------------------------------- /utils/nms/gpu_nms.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms/gpu_nms.hpp -------------------------------------------------------------------------------- /utils/nms/gpu_nms.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms/gpu_nms.pyx -------------------------------------------------------------------------------- /utils/nms/nms_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms/nms_kernel.cu -------------------------------------------------------------------------------- /utils/nms/py_cpu_nms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms/py_cpu_nms.py -------------------------------------------------------------------------------- /utils/nms_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/nms_wrapper.py -------------------------------------------------------------------------------- /utils/plot_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/plot_loss.py -------------------------------------------------------------------------------- /utils/pycocotools/UPSTREAM_REV: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/UPSTREAM_REV -------------------------------------------------------------------------------- /utils/pycocotools/__init__.py: -------------------------------------------------------------------------------- 1 | __author__ = 'tylin' 2 | -------------------------------------------------------------------------------- /utils/pycocotools/_mask.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/_mask.c -------------------------------------------------------------------------------- /utils/pycocotools/_mask.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/_mask.pyx -------------------------------------------------------------------------------- /utils/pycocotools/coco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/coco.py -------------------------------------------------------------------------------- /utils/pycocotools/cocoeval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/cocoeval.py -------------------------------------------------------------------------------- /utils/pycocotools/license.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/license.txt -------------------------------------------------------------------------------- /utils/pycocotools/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/mask.py -------------------------------------------------------------------------------- /utils/pycocotools/maskApi.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/maskApi.c -------------------------------------------------------------------------------- /utils/pycocotools/maskApi.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/pycocotools/maskApi.h -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/vis_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/vis_util.py -------------------------------------------------------------------------------- /utils/yolo.c: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/yolo.c -------------------------------------------------------------------------------- /utils/yolo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/yolo.py -------------------------------------------------------------------------------- /utils/yolo.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/yolo.pyx -------------------------------------------------------------------------------- /utils/yolo_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/utils/yolo_v2.py -------------------------------------------------------------------------------- /yolo_detect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cory8249/yolo2-pytorch/HEAD/yolo_detect.py --------------------------------------------------------------------------------