-
Notifications
You must be signed in to change notification settings - Fork 19
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Error with yolo-v3(416*416) #1
Comments
I found that the filters of 255 should be changed to 18 in my case. Thanks. |
you must generate yolov3-416.onnx by change the cfg file and yolov3_to_onnx.py file ,the height and width and the out shape 19,38,76 to 13,26,52, I change this, and success! |
@Hanlos hello, thanks for you sharing, but how can i change the yolov3_to_onnx.py file? can you provide guidance, thanks. |
Hello,I have a similar problem when I convert my weight, I changed the cfg and trt_yolo3_module_1batch.py `[convolutional] [yolo]
|
Did you test the project with yolo-v3(416*416)?
$ python3 trt_yolo3_module_1batch.py
Reading engine from file yolov3-608.trt
Traceback (most recent call last):
File "trt_yolo3_module_1batch.py", line 214, in
output_dic_list = alpha_yolo3_unit.process_frame_batch(input_dic_list)
File "trt_yolo3_module_1batch.py", line 185, in process_frame_batch
(class_list_all,box_list_all,conf_list_all) = self.detection(procession_tuple)
File "trt_yolo3_module_1batch.py", line 99, in detection
output = output.reshape(shape)
ValueError: cannot reshape array of size 3042 into shape (1,255,13,13)
The text was updated successfully, but these errors were encountered: