Evaluation of YOLO Models on DIOR Dataset

Authors

• Amulya Harihara Narayana Rao
• Vigneshwar Karuppiah Ramanathan

Dataset

The dataset used in our experiments is THIS DATASET.

1. DIOR dataset

   • Exploratory Data Analysis on Dior Dataset

   • Annotation formatting

   • Bounding box representation of ground truth

2. Under the Dior directory, you can also do the following:

   • Yolo annotation converter
   • Dataset splitting
   • Data Augmentation
   • File name extraction
   • Plot sample
   • Sanity check

Model(s)

The models employed in our experiments were YOLOv8 and YOLO-NAS.

Results

Results YOLOv8:

Results YOLO-NAS: