search

theAIGuysCode / yolov4-custom-functions

A Wide Range of Custom Functions for YOLOv4, YOLOv4-tiny, YOLOv3, and YOLOv3-tiny Implemented in TensorFlow, TFLite, and TensorRT.
MIT License
602 stars 370 forks source link

error when save pre-train mode to tensor flow #89

Closed linuxfedora2020 closed 2 years ago

linuxfedora2020 commented 2 years ago

called: python save_model.py --weights ./data/yolov4.weights --output ./checkpoints/yolov4-416 --input_size 416 --model yolov4 with the error below:

Traceback (most recent call last): File "save_model.py", line 58, in app.run(main) File "/home/xb/anaconda3/envs/yolov4-gpu/lib/python3.7/site-packages/absl/app.py", line 312, in run _run_main(main, args) File "/home/xb/anaconda3/envs/yolov4-gpu/lib/python3.7/site-packages/absl/app.py", line 258, in _run_main sys.exit(main(argv)) File "save_model.py", line 54, in main save_tf() File "save_model.py", line 32, in save_tf for i, fm in enumerate(feature_maps):