A detector for any kinds of videos or images. Using YOLOv5 models and PyTorch for object detection.
If running on Localhost, please ensure that the PyTorch has been installed in the environment.
- Image detector
python YOLOv5s2.py # change to your image path
- Video detector
python YOLOv5s_video_.py # change to your video path
Select a video or an image, and modify the original path to your path. Then, the output image will be in runs/detect