├── .gitignore ├── LICENSE ├── README.md ├── dataset └── kitti │ └── ImageSets │ ├── test.txt │ ├── train.txt │ ├── trainval.txt │ └── val.txt ├── docs ├── demo.gif └── yolov4_architecture.png ├── requirements.txt └── src ├── config ├── __init__.py ├── cfg │ └── yolo3d_yolov4.cfg ├── kitti_config.py └── train_config.py ├── data_process ├── __init__.py ├── kitti_bev_utils.py ├── kitti_data_utils.py ├── kitti_dataloader.py ├── kitti_dataset.py ├── train_val_split.py └── transformation.py ├── evaluate.py ├── models ├── __init__.py ├── darknet2pytorch.py ├── darknet_utils.py ├── model_utils.py └── yolo_layer.py ├── test.py ├── test.sh ├── train.py ├── train.sh └── utils ├── __init__.py ├── cal_intersection_rotated_boxes.py ├── evaluation_utils.py ├── find_anchors.py ├── iou_rotated_boxes_utils.py ├── logger.py ├── misc.py ├── torch_utils.py ├── train_utils.py └── visualization_utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /dataset/kitti/ImageSets/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/dataset/kitti/ImageSets/test.txt -------------------------------------------------------------------------------- /dataset/kitti/ImageSets/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/dataset/kitti/ImageSets/train.txt -------------------------------------------------------------------------------- /dataset/kitti/ImageSets/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/dataset/kitti/ImageSets/trainval.txt -------------------------------------------------------------------------------- /dataset/kitti/ImageSets/val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/dataset/kitti/ImageSets/val.txt -------------------------------------------------------------------------------- /docs/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/docs/demo.gif -------------------------------------------------------------------------------- /docs/yolov4_architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/docs/yolov4_architecture.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/config/cfg/yolo3d_yolov4.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/config/cfg/yolo3d_yolov4.cfg -------------------------------------------------------------------------------- /src/config/kitti_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/config/kitti_config.py -------------------------------------------------------------------------------- /src/config/train_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/config/train_config.py -------------------------------------------------------------------------------- /src/data_process/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/data_process/kitti_bev_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/kitti_bev_utils.py -------------------------------------------------------------------------------- /src/data_process/kitti_data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/kitti_data_utils.py -------------------------------------------------------------------------------- /src/data_process/kitti_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/kitti_dataloader.py -------------------------------------------------------------------------------- /src/data_process/kitti_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/kitti_dataset.py -------------------------------------------------------------------------------- /src/data_process/train_val_split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/train_val_split.py -------------------------------------------------------------------------------- /src/data_process/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/data_process/transformation.py -------------------------------------------------------------------------------- /src/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/evaluate.py -------------------------------------------------------------------------------- /src/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/models/darknet2pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/models/darknet2pytorch.py -------------------------------------------------------------------------------- /src/models/darknet_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/models/darknet_utils.py -------------------------------------------------------------------------------- /src/models/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/models/model_utils.py -------------------------------------------------------------------------------- /src/models/yolo_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/models/yolo_layer.py -------------------------------------------------------------------------------- /src/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/test.py -------------------------------------------------------------------------------- /src/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/test.sh -------------------------------------------------------------------------------- /src/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/train.py -------------------------------------------------------------------------------- /src/train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/train.sh -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/utils/cal_intersection_rotated_boxes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/cal_intersection_rotated_boxes.py -------------------------------------------------------------------------------- /src/utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/evaluation_utils.py -------------------------------------------------------------------------------- /src/utils/find_anchors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/find_anchors.py -------------------------------------------------------------------------------- /src/utils/iou_rotated_boxes_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/iou_rotated_boxes_utils.py -------------------------------------------------------------------------------- /src/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/logger.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/torch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/torch_utils.py -------------------------------------------------------------------------------- /src/utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/train_utils.py -------------------------------------------------------------------------------- /src/utils/visualization_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/maudzung/YOLO3D-YOLOv4-PyTorch/HEAD/src/utils/visualization_utils.py --------------------------------------------------------------------------------