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
604 stars 369 forks source link

How to make custom one (the car plate detection model) to tensorrt? #72

Open edwardchang0112-nctu opened 3 years ago

edwardchang0112-nctu commented 3 years ago

I followed the instructions on the post The steps:

python3 save_model.py --weights ./data/custom.weights --output ./checkpoints/custom.tf --input_size 416 --model yolov4
python3 convert_trt.py --weights ./checkpoints/custom.tf --quantize_mode float16 --output ./checkpoints/custom-trt-fp16-416

but got the error:

Traceback (most recent call last):
  File "convert_trt.py", line 100, in <module>
    app.run(main)
  File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 303, in run
    _run_main(main, args)
  File "/usr/local/lib/python3.6/dist-packages/absl/app.py", line 251, in _run_main
    sys.exit(main(argv))
  File "convert_trt.py", line 96, in main
    save_trt()
  File "convert_trt.py", line 58, in save_trt
    max_batch_size=8)
  File "<string>", line 28, in _replace
ValueError: Got unexpected field names: ['max_batch_size']

Any idea on this issue?

animeesh commented 2 years ago

i got the same. do let me know if you got the solution

farshokat commented 2 years ago

i need your answer, thanks