-
-
Notifications
You must be signed in to change notification settings - Fork 17.2k
Closed
Labels
questionFurther information is requestedFurther information is requested
Description
Search before asking
- I have searched the YOLOv5 issues and discussions and found no similar questions.
Question
It seems that loading the classification model can't leave out of the YOLOV5-Classifier code base. When loading the model,
model = torch.load('path/to/best.pt', map_location=torch.device('cpu'))['model'].float()
, it throws error like this
"""
module = self._system_import(name, *args, **kwargs)
ModuleNotFoundError: No module named 'models'
"""
Once the folders of models
and utils
are copied to a new project directory, the loading works.
I am asking if there is a way I can load the classification model and use it as a stand-alone and for example, converting to TensorFlow like the object detection codebase.
As I know, the YOLOV5 object detection doesn't have this problem.
Thanks.
Additional
No response
Metadata
Metadata
Assignees
Labels
questionFurther information is requestedFurther information is requested