Replies: 1 comment 1 reply
-
Do a prediction first, then select your class. |
Beta Was this translation helpful? Give feedback.
1 reply
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I am calling yolo yolov5m.pt model with this code. I only want to detect the people from my environment. For this aim what should I add to this line ?
model = torch.hub.load("./yolov5", 'custom', path="./yolov5/yolov5m.pt", source='local')
I tried with this :
model = torch.hub.load("./yolov5", 'custom', path="./yolov5/yolov5m.pt", source='local' , classes = 0 )
and then I got TypeError: custom() got an unexpected keyword argument 'classes' error. Is it not possible to select the detected class, if we work with custom model?Beta Was this translation helpful? Give feedback.
All reactions