yolov4无法导出模型
Created by: thunder95
!python tools/export_model.py -c myconfigs/yolov4_cspdarknet_voc.yml --output_dir=export_model/v1 -o weights=output/yolov4_cspdarknet_voc/best_model TestReader.inputs_def.image_shape=[3,608,608]
/home/aistudio/work/PaddleDetection Traceback (most recent call last): File "tools/export_model.py", line 215, in main() File "tools/export_model.py", line 196, in main test_fetches = model.test(feed_vars) File "/home/aistudio/work/PaddleDetection/ppdet/modeling/architectures/yolo.py", line 162, in test return self.build(feed_vars, mode='test') File "/home/aistudio/work/PaddleDetection/ppdet/modeling/architectures/yolo.py", line 90, in build im_size = feed_vars['im_size'] KeyError: 'im_size'
不知道这个im_size该怎么配置