Skip to content

In this project a neural net learns to draw the PyTorch logo using Tensorboard, Pytorch, Numpy, and Houdini

Notifications You must be signed in to change notification settings

walabe8/Draw-Logo_ML-Regression-Model

Repository files navigation

Draw Logo - ML Regression Model

PROCESS

  1. Create training data in Houdini
  2. Build and train our own neural nets with Python and PyTorch
  3. Integrate them into our Houdini Workflows using ONNX data

PROJECT

  • In this project a neural net learns to draw the PyTorch logo.
  • To do so we create our own training data in Houdini, we build and train our own neural nets with Python and PyTorch and integrate them into a Houdini workflow using ONNX.
  • The logo is fed into Houdini in where we convert the logo to 50,000 points and convert the logo to a float value image.
    • output_a output_b represent the X and Y coordinates and our target data is what we want our neural network to learn to draw (the logo)
  • This raw data is fed to our neural network we create with Pytorch and Tensorboard provides a much better judgement over how well our model.
Pytorch_Logo_Tensoboard
  • The training data is fed back into Houdini and ONNX data is interpreted and rendered.
Pytorch_Logo_50 Pytorch_Logo_100 Pytorch_Logo_250 Pytorch_Logo_500 Pytorch_Logo_1000



Houdini Snippets input_a input_b target ONNX_data

TOOLS

About

In this project a neural net learns to draw the PyTorch logo using Tensorboard, Pytorch, Numpy, and Houdini

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages