search

ubc-subbots / triton

ROS2 system for the Triton AUV
11 stars 1 forks source link

Upgrade training pipeline to support YOLOv4 #98

Open kevinh42 opened 2 years ago

kevinh42 commented 2 years ago

Currently, we are using YOLOv3, but YOLOv4 provides better detection. triton_object_recognition and the training pipeline will need to be updated to support YOLOv4. This will also require upgrading to OpenCV 4.5.3 since 4.2 doesn't support YOLOv4.