thnak/yolov7-2-tensorflow

export any your YOLOv7 model to TensorFlow, TensorFlowJs, ONNX, OpenVINO, RKNN,...

yolov7-2-tensorflow's Issues

TFlite export failed

Hi, thanks for sharing this amazing project.

However, I encounter an error when I try to export the yolov7.pt to tflite format by command line:

python export.py --weights "yolov7.pt" --include tflite

The error message is as below:

Traceback (most recent call last):
File "/content/yolov7-2-tensorflow/export.py", line 451, in
outputpath, s_models = export_saved_model(model_ori,
File "/content/yolov7-2-tensorflow/tools/auxexport.py", line 104, in export_saved_model
tf_model = TFModel(cfg=model.yaml, model=model.cpu(), nc=len(model.names), imgsz=imgsz)
File "/content/yolov7-2-tensorflow/models/tf.py", line 602, in init
self.model, self.savelist = parse_model(deepcopy(self.yaml), ch=[ch], model=model, imgsz=imgsz)
File "/content/yolov7-2-tensorflow/models/tf.py", line 575, in parse_model
else tf_m(*args, w=model.model[i]) # module
File "/content/yolov7-2-tensorflow/models/tf.py", line 72, in init
[TFConv2d(c1, c2, k, s, g=g, bias=False, w=w.rbr_dense[0]),
TypeError: 'NoneType' object is not subscriptable

Did I miss something?
Thank you again!

File not found: mydataset.yaml

Hi,
thanks for sharing your work, this repo is really useful! I found an issue when I try to use --int8. Any help would be appreciated!

python3 export.py --include tflite --int8 --weights ./yolov7-tiny.pt
....
Tensorflow lite: starting export with tensorflow 2.8.0...
Tensorflow lite: export failure❌:
File not found: mydataset.yaml
....