Issue when I run python object_detection_demo_yolov3_async.py

[nitipol012]

I've converted my custom data on the YoloV4 model it's worked. but when I run python object_detection_demo_yolov3_async.py I got this error

[ INFO ] Creating Inference Engine... [ INFO ] Loading network [ INFO ] Preparing inputs [ INFO ] Loading model to the plugin [ INFO ] Starting inference... To close the application, press 'CTRL+C' here or switch to the output window and press ESC key To switch between min_latency/user_specified modes, press TAB key in the output window Traceback (most recent call last): File "object_detection_demo_yolov3_async.py", line 518, in sys.exit(main() or 0) File "object_detection_demo_yolov3_async.py", line 398, in main args.keep_aspect_ratio) File "object_detection_demo_yolov3_async.py", line 233, in get_objects out_blob = out_blob.buffer.reshape(net.layers[net.layers[layer_name].parents[0]].out_data[0].shape) ValueError: cannot reshape array of size 689520 into shape (1,21,52,52)

I used image size when I'm training is 416x416x3

Thank you

[Usernamezhx]

maybe you should change the yolo_v4_tiny.json classes num

[nitipol012]

Excellent. I follow your comment. It's worked. Thanksyou