Open jccmoing-bit opened 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')
@jccmoing-bit how to pass in custom names
Unable to fix the issue.
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](https://user-images.githubusercontent.com/72694831/174226835-da6fc795-ba1d-4dcc-bd07-5555d5014d19.jpg)