Export yolov3 model with shape of weights error
Created by: lyw615
command : python tools/export_model.py -c configs/yolov3_darknet.yml --output_dir=./inference_model -o weights=../model/car_p/vehicle_yolov3_darknet YoloTestFeed.image_shape=[3,608,608]
logs::2019-11-27 15:09:45,628-INFO: Loading parameters from ../model/car_p/vehicle_yolov3_darknet... Traceback (most recent call last): File "tools/export_model.py", line 120, in main() File "tools/export_model.py", line 107, in main checkpoint.load_params(exe, infer_prog, cfg.weights) File "./ppdet/utils/checkpoint.py", line 118, in load_params fluid.io.load_vars(exe, path, prog, predicate=_if_exist) File "/software/conda/envs/super_mask/lib/python3.6/site-packages/paddle/fluid/io.py", line 682, in load_vars filename=filename) File "/software/conda/envs/super_mask/lib/python3.6/site-packages/paddle/fluid/io.py", line 741, in load_vars format(orig_shape, each_var.name, new_shape))
**RuntimeError: Shape not matching: the Program requires a parameter with a shape of ((255, 1024, 1, 1)), while the loaded parameter (namely [ yolo_output.0.conv.weights ]) has a shape of ((33, 1024, 1, 1))
The weights can be used to infer image correctly,but with error in model export. Is the code or config file not updated?
Reference from: https://github.com/PaddlePaddle/PaddleDetection/blob/release/0.1/docs/EXPORT_MODEL.md