itsekiri-tatashe/nigerian-headwear

Detecting Native Headwear worn in Nigeria using Computer Vision # Nigerian Native Headwear Detection: Project Overview ___ The purpose of this project was to use Object Detection (Computer Vision) detect some of the most popular traditional headgears worn in Nigeria namely; * Abeti Aja from the South West, * Awolowo from the South West, * Fila from the South West, * Gele from the South West; * Fula from the North; * Okpu-Ozo from the South-East * Okpu-Agu from the South-East * the Turban ___ * Used **Google Web Scraper** to collect the data/images * Performed data-label annonatation using **Label Studio** * Used **YoloV5** to train **Object Detection** model on the dataset ___ ## Model Performamce #### Confusion Matrix #### F1_curve #### PR_curve ___ ## How to Detect Images * Clone repo and install requirements.txt in a **Python>=3.7.0** environment, including **PyTorch>=1.7**. ```bash git clone github.com # clone cd yolov5 pip install -r requirements.txt # install ``` * Add the images or videos you want to detect in the ```detect-img/``` folder * Run this command in your terminal ```bash python detect.py --weights best.pt --source detect-img/img-name.jpg detect-img/img-video.mp4 'youtu.be' # YouTube 0 #Webcam ``` * Go to the `runs/detect/` folder to see the results!