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)

  ---

The dataset used is available on Kaggle.

Usage

1. Clone the repository

git clone https://github.com/RsGoksel/Military-Vehicles-Detection

2. Install requirements

pip install -r requirements.txt

3. Replace your images to the source folder

4. Run the model

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

Train

1. Clone the repository

git clone https://github.com/RsGoksel/Military-Vehicles-Detection

2. Install requirements

pip install -r requirements.txt

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

4. Train a model

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