Skip to content

Latest commit

 

History

History
61 lines (38 loc) · 1.36 KB

README.md

File metadata and controls

61 lines (38 loc) · 1.36 KB

Military Vehicles Image Recognition

Military Vehicles Object detection from ARMA Game

The model is able to detect these classes

  • CSAT Varsuk
  • CSAT Marid
  • CSAT Zamak (Transport)

test_01

The dataset used is available on Kaggle.

Usage

  1. Clone the repository
git clone https://github.com/RsGoksel/Military-Vehicles-Detection
  1. Install requirements
pip install -r requirements.txt

  1. Replace your images to the source folder

  2. Run the model

python detect.py --source ./input/ --weights runs/train/yolo_tr/weights/best.pt --conf 0.5 --name Arma_Detection

Gif

Train

  1. Clone the repository
git clone https://github.com/RsGoksel/Military-Vehicles-Detection
  1. Install requirements
pip install -r requirements.txt

  1. Execute Data handling file if you got "xml" file format dataset version. If you downloaded mine, just create train-val folders.

  2. Train a model

python train.py --data './data.yaml' --epochs 10