search

pooya-mohammadi / yolov5-gradcam

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

For Customize dataset #15

Open noreenanwar opened 2 years ago

noreenanwar commented 2 years ago

Can I use it for customize data?

pooya-mohammadi commented 2 years ago

@noreenanwar This code only needs the saved model. The model could be trained on a custom dataset or one of the yolov5's standard models.