This repository contains a PyTorch implementation of ComplexYOLO using YOLO version 4. For an implementation using YOLOv3 see here. It is build to be applied on the data from the Astyx Dataset. For an implementation for the KITTI dataset see here.
pip install -U -r requirements.txt
- Install all requirements
- Download or clone this repo by using
git clone https://github.com/BerensRWU/AstyxComplexYOLOv4/
in the terminal. - Save the Astyx dataset in the folder
dataset
.(See Section Astyx HiRes). - Download the weights for the RADAR and LiDAR detector from the moodle page of the Lecture.
The Astyx HiRes is a dataset from Astyx for object detection for autonomous driving. Astyx has a sensor setup consisting of camera, LiDAR, RADAR. Additional information can be found here: Dataset Paper and Specification
└── dataset/
├── dataset_astyx_hires2019 <-- 546 data
| ├── calibration
| ├── camera_front
| ├── groundtruth_obj3d
| ├── lidar_vlp16
└── ├── radar_6455