search

theAIGuysCode / Object-Detection-API

Yolov3 Object Detection implemented as APIs, using TensorFlow and Flask
Apache License 2.0
339 stars 179 forks source link

ImportError: No module named yolov3_tf2.models #15

Open Milan-Koelen opened 4 years ago

Milan-Koelen commented 4 years ago

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

netvor-73 commented 4 years ago

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