Error exec command python2 save_model.py --weights ./data/yolov4-tiny.weights --output ./checkpoints/yolov4-tiny-416 --input_size 416 --model yolov4 --tiny

hunglc007 / tensorflow-yolov4-tflite

YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite

https://github.com/hunglc007/tensorflow-yolov4-tflite

MIT License

2.24k stars 1.24k forks source link

Error exec command python2 save_model.py --weights ./data/yolov4-tiny.weights --output ./checkpoints/yolov4-tiny-416 --input_size 416 --model yolov4 --tiny #297

Open yoooco opened 3 years ago

yoooco commented 3 years ago

Hello here, I hope someone can help me, thanks in advance. This is my development environment :

Ubuntu 20.04 (VirtualBox, 3CPU)
OpenCV 4.2.0
Tensorflow 2.3.0rc0

When i run : python2 save_model.py --weights ./data/yolov4-tiny.weights --output ./checkpoints/yolov4-tiny-416 --input_size 416 --model yolov4 --tiny

i got this Error : Traceback (most recent call last): File "save_model.py", line 4, in from core.yolov4 import YOLO, decode, filter_boxes ImportError: No module named core.yolov4

preetd14 commented 3 years ago

Hey did you find a fix for this? I am getting the same error

ghost commented 3 years ago

Please try to create the __init__.py file in both tensorflow-yolov4-tflite folder, and all of its subdirectories.

In Linux, for example, use the following command touch __init__.py (just a blank file, you don't need to edit anything)
In Windows, just create a new file .txt, rename it and its extension to __init__.py and copy this file to all subfolders.

I hope this help <3 Have a good conversion!