YOLO-ELA: Efficient Local Attention Modeling for High-Performance Real-Time Insulator Defect Detection
Install requirements
pip install -r requirements.txt
- +NVIDIA GPU + CUDA CuDNN
- +Linux (Ubuntu)
- +Python 3.10
- Download YOLO-ELA checkpoint here
- Open a terminal and run
python test.py\
--model 'models/yolo_ela.pt'\
This automatically create a new directory called run
. Navigate to see results
- Open terminal and run
python train.py \
--model_scale 'ela' \
--cfg 'ultralytics/cfg/data.yaml' \
--aug True \
--name 'ela' \
--epochs 100 \
--bs 16 \
--img_sz 640 # Image size can either be 320 or 640
@article{yoloela,
author = {Olalekan Akindele and Joshua Atolagbe},
title = {YOLO-ELA: Efficient Local Attention Modeling for High-Performance Real-Time Insulator Defect Detection},
journal = {arXiv preprint arXiv:2410.11727},
year = {2024}
}
The codes in this repository are based on Ultralytics