Dataset

• CHVG: https://figshare.com/articles/dataset/CHVG_Dataset/19625166
• CHV: https://www.mdpi.com/1424-8220/21/10/3478
• Roboflow: https://universe.roboflow.com/hx-hezqh/ppe-detection-yfmym

Custom Dataset

• Require labeling and annotating

install pytorch

pip3 install torch torchvision torchaudio --index-url <https://download.pytorch.org/whl/cu121>

setup environment

pip install virtualenv

python -m venv yolo_nas_env

#activate script
yolo_nas_env\\Scripts\\activate.bat

install super gradient

pip install super-gradients supervision

inside dataset folder

pip install simple_image_download==0.4

from simple_image_download import simple_image_download as simp
#create a new class instance
response = simp.simple_image_download

#load images downloaded from google
keywords = ["helmets", "boots", "gloves", "humans", "vests" ]
# for each key word download images
for kw in keywords:
	response().download(kw,200)

python dataset\\image.py

Annotate the image