search

pooya-mohammadi / yolov5-gradcam

Visualizing Yolov5's layers using GradCam
MIT License
281 stars 47 forks source link

make --img-path parameter accept folder as a path #10

Closed pourmand1376 closed 2 years ago

pourmand1376 commented 2 years ago

I added a feature so that --img-path can accept folder as path. So, if it is folder, it will loop through all files and interpret them one by one.

This is default functionality in YOLO.

pooya-mohammadi commented 2 years ago

Thank you for your pull request. It's done!