ONNX export variable batch-size

[daixiangzi]

No description provided.

[daixiangzi]

your code only support bs =1,

[daixiangzi]

i should modify it ,make it suport different batch_szie onnx export

[glenn-jocher]

@daixiangzi yes it is hard coded to batch-size 1. If you'd like to submit a PR for multiple batch sizes we can look over it!

One thing to keep in mind is that all of the shapes and constants must be fixed in the model forward pass to prevent onnx errors, so if you make batch size a variable you should probably define it in the model init as a constant rather than using p.shape as in the default forward method.

[daixiangzi]

@daixiangzi yes it is hard coded to batch-size 1. If you'd like to submit a PR for multiple batch sizes we can look over it!

One thing to keep in mind is that all of the shapes and constants must be fixed in the model forward pass to prevent onnx errors, so if you make batch size a variable you should probably define it in the model init as a constant rather than using p.shape as in the default forward method.

yes，Only bs=fixed_number,Not bs,,,_ = p.shape，i have completed multiple batch size

[daixiangzi]

@daixiangzi yes it is hard coded to batch-size 1. If you'd like to submit a PR for multiple batch sizes we can look over it!

One thing to keep in mind is that all of the shapes and constants must be fixed in the model forward pass to prevent onnx errors, so if you make batch size a variable you should probably define it in the model init as a constant rather than using p.shape as in the default forward method.

i haved submit pr,but it look like failed,i don't konw what's the reason?

[glenn-jocher]

Ok, I’ll try to take a look tomorrow! Have you exported to onnx successfully using the pr code?

[daixiangzi]

Ok, I’ll try to take a look tomorrow! Have you exported to onnx successfully using the pr code?

yes，i can export different fixed batch size onnx model,but it appear error,when i try to export dynamic batch model,

[github-actions]

This issue is stale because it has been open 30 days with no activity. Remove Stale label or comment or this will be closed in 5 days.

[bing1zhi2]

Ok, I’ll try to take a look tomorrow! Have you exported to onnx successfully using the pr code?

yes，i can export different fixed batch size onnx model,but it appear error,when i try to export dynamic batch model,
@daixiangzi
Thank your for your job....Any update? I'm trying to export a batch size 32 onnx model, but the output node don't have the batch dimension.

[luvwinnie]

any progress on this? I'm trying to export a variable batch size for onnx model, suppose with the dynamic_axes can be dynamic, but it seems like not working.

[glenn-jocher]

@luvwinnie thanks for your patience! The YOLOv3 model has strict shape requirements for exporting to ONNX, especially for dynamic batch sizes. We’re making progress on adding support for exporting models with dynamic batch sizes and hope to have an update soon. Thank you for your understanding and dedication to improving the YOLOv3 model.