Skip to content
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

onnx export #225

Closed
123liluky opened this issue Jun 29, 2020 · 2 comments
Closed

onnx export #225

123liluky opened this issue Jun 29, 2020 · 2 comments
Labels

Comments

@123liluky
Copy link

123liluky commented Jun 29, 2020

lu@deeplearningHost2:/data/lilu/yolov5-master/yolov5-master$ python models/onnx_export.py
Namespace(batch_size=1, img_size=[640, 640], weights='./weights/test.pt')
Fusing layers...
Model Summary: 140 layers, 1.81977e+06 parameters, 1.65722e+06 gradients, 3.3 GFLOPS
graph torch-jit-export (
%images[FLOAT, 1x3x640x640]
) initializers (
%661[INT64, 1]
%662[INT64, 1]
%663[INT64, 1]
%664[INT64, 1]
%665[INT64, 1]
%666[INT64, 1]
%model.0.conv.conv.bias[FLOAT, 16]
%model.0.conv.conv.weight[FLOAT, 16x12x3x3]
%model.1.conv.bias[FLOAT, 32]
%model.1.conv.weight[FLOAT, 32x16x3x3]
%model.10.conv.bias[FLOAT, 128]
%model.10.conv.weight[FLOAT, 128x256x1x1]
%model.13.bn.bias[FLOAT, 128]
%model.13.bn.running_mean[FLOAT, 128]
%model.13.bn.running_var[FLOAT, 128]
%model.13.bn.weight[FLOAT, 128]
......
When I export .onnx,some Int64 parameters appears. I am doubt what are they? Thank you.

@github-actions
Copy link
Contributor

github-actions bot commented Jun 29, 2020

Hello @123liluky, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook Open In Colab, Docker Image, and Google Cloud Quickstart Guide for example environments.

If this is a bug report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom model or data training question, please note that Ultralytics does not provide free personal support. As a leader in vision ML and AI, we do offer professional consulting, from simple expert advice up to delivery of fully customized, end-to-end production solutions for our clients, such as:

  • Cloud-based AI systems operating on hundreds of HD video streams in realtime.
  • Edge AI integrated into custom iOS and Android apps for realtime 30 FPS video inference.
  • Custom data training, hyperparameter evolution, and model exportation to any destination.

For more information please visit https://www.ultralytics.com.

@github-actions
Copy link
Contributor

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant