Can I convert this pytorch version to tensorflow ?? Will the tensorflow version run with the same speed of training or inference??

[dorecasan]

🚀 Feature

Motivation

Pitch

Alternatives

Additional context

[github-actions]

Hello @dorecasan, thank you for your interest in our work! Please visit our Custom Training Tutorial to get started, and see our Jupyter Notebook , 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.

[glenn-jocher]

@dorecasan yes, I believe it should be possible to recreate similar training and inference results in TF.

[github-actions]

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.

[faldisulistiawan]

@dorecasan Hello, have you been able to convert Yolov5 Pytorch to Tensorflow? If yes, can you please give me the tutorial for it? Thank you

[glenn-jocher]

@faldisulistiawan @dorecasan yes YOLOv5 is convertible to TF2 and exportable to TFLite. See #1127

[faldisulistiawan]

@glenn-jocher Thank you so much for the info! I needed the TF for future production needs, since productionizing TF models is easier. Although, #1127 can only be used as the inference / prediction right? I still need to train the yolov5 in the pytorch framework?

[zldrobit]

@faldisulistiawan Yes, you still have to train in PyTorch framework.
#1127 is only for exporting trained models.

[faldisulistiawan]

@zldrobit Thanks!