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.

Sign up for GitHub

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

Jump to bottom

[Solved] How did you transform the pytorch model to TFJS? #4

Open
Pensarfeo opened this issue Feb 6, 2023 · 2 comments
Open

[Solved] How did you transform the pytorch model to TFJS? #4

Pensarfeo opened this issue Feb 6, 2023 · 2 comments

Comments

@Pensarfeo
Copy link

Hi, can you give more info about how you transformed the pytorch graph to a tfjs one? I tried but I run over tons of errors.

Thanks and great work!
@hugozanini
Copy link
Owner

Hi @Pensarfeo! I wrote an article detailing the pipeline for training the model in PyTorch and converting it to TensorFlow.js. Please let me know if it helps :)

https://towardsdatascience.com/training-a-custom-yolov7-in-pytorch-and-running-it-directly-in-the-browser-with-tensorflow-js-96a5ecd7a530

@hugozanini hugozanini changed the title How did you transform the pytorch model to TFJS? [Solved] How did you transform the pytorch model to TFJS? Mar 30, 2023
@Pensarfeo
Copy link
Author

@hugozanini
Thanks for sharing, I see you use it real time, do you know what is your FPS?

Yours
Pedro

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@Pensarfeo @hugozanini