Skip to content

iremustek/uav-object-detection-yolov5x

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 

Repository files navigation

UAV Object Detection using Transfer Learning with YOLOv5x Open In Colab

This project focuses on performing UAV object detection using transfer learning with YOLOv5x. The work is conducted on Google Colab, utilizing a GPU for accelerated computations (NVIDIA-SMI 525.85.12, CUDA Version: 12.0).

 Link

Dataset

The dataset consists of the following components:

  • images: A collection of 314 diverse drone images.
  • labels: Each image is annotated with an ID and bounding box information, specifying the position and dimensions of detected objects in the images (format: [ID, x, y, width, height]).

Folder Tree

After the train-test-validation split is done, the overall folder tree will be like that:

└── uav-object-detection-yolov5x /
    ├── data/
    │   ├── images/
    │   │   ├── train
    │   │   ├── test
    │   │   └── val
    │   └── labels/
    │       ├── train
    │       ├── test
    │       └── val   
    └── uav-object-detection-yolov5x.ipynb

About

UAV Object Detection using transfer learning with YOLOv5x

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published