theAIGuysCode/Object-Detection-API

ImportError: No module named yolov3_tf2.models

Opened this issue · 1 comments

Hey guys, I'm trying to convert my custom weights files to TensorFlow format but keep running into the same error.
I've renamed my files to yolov3.weights and placed then in the same folder as the official pretrained weights. I've also changed the coco.names file accordingly.
I have also changed the number of classes to match in the app.py, detect_video.pi and detect.py
Please let me know what I'm doing wrong!

(yolov3-cpu) MacBook-Pro-Milan-2:Object-Detection-API Milan$ python load_weights.py Traceback (most recent call last): File "load_weights.py", line 4, in <module> from yolov3_tf2.models import YoloV3, YoloV3Tiny ImportError: No module named yolov3_tf2.models

Milan

the folder yolov3_tf2 has to be in the same location as load_weights.py