Fine-Tuning YOLOv5 to detect Military Vehicles in Aerial ARMA 3 Imagery
A Youtube devlog of the project is available here:
Currently, the model is able to detect the following classes:
- CSAT Varsuk
- CSAT Marid
- CSAT Zamak (Transport)
The model is a YOLOv5 fine tuned using 100 images of each class using various environments and angles at noon clear sky using a UAV at around 100 meters altitude.
The dataset used is available on Kaggle.
- Clone the repository
git clone https://github.com/AlexandreSajus/Military-Vehicles-Image-Recognition.git- Install the requirements
pip install -r requirements.txt-
Add your images to the
inputfolder -
Run the model
python detect.py --source ./input/ --weights runs/train/yolo_arma4/weights/best.pt --conf 0.5 --name yolo_arma- The results will be available in a
runs/detect/yolo_armaXfolder