search

pooya-mohammadi / yolov5-gradcam

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

Label problem #16

Open jccmoing-bit opened 2 years ago

jccmoing-bit commented 2 years ago

I trained the model on my own dataset, its category is not the same as the coco dataset, but the final display result seems to be the label of the coco dataset? Japan_008228-res

pooya-mohammadi commented 2 years ago

@jccmoing-bit Hi jccmoin, check the main.py for more help. There is an input argument by the name name.

...
parser.add_argument('--names', type=str, default=None,
                    help='The name of the classes. The default is set to None and is set to coco classes. Provide your custom names as follow: object1,object2,object3')
pooya-mohammadi commented 2 years ago

@jccmoing-bit how to pass in custom names

mohaliyet commented 1 year ago

image Unable to fix the issue.