Polygon Yolov5 with adjustments to enable polygon prediction boxes
Polygon-Yolov5 This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes. Section I. Description The codes are based on Ultralytics/yolov5, and several functions are added and modified to enable polygon prediction boxes. The modifications compared with Ultralytics/yolov5 and their brief descriptions are summarized below: data/polygon_ucas.yaml : Exemplar UCAS-AOD dataset to test the effects of polygon boxes data/images/UCAS-AOD : For the inference of polygon-yolov5s-ucas.pt models/common.py :3.1. class Polygon_NMS : Non-Maximum Suppression (NMS) module for Polygon Boxes3.2. class Polygon_AutoShape […]
Read more