Bismuth-Consultancy-BV / MLOPs

Machine Learning Toolset for Houdini
BSD 3-Clause "New" or "Revised" License
371 stars 52 forks source link

Image Segmentation Pipeline and Image to 3d #149

Open pedromartello opened 5 months ago

pedromartello commented 5 months ago

This PR introduces three nodes to MLOPs. An Image to 3d model processor based on TripoSR, and an image segmentation pipeline (and related image segment) node that can use image segmentation models from hugging face. Usage examples are in hip/ToBeValidated/image_segmentation_and_3d.hip

In a preliminary commit, It also introduces a multistage requirements.txt file so that packages that are compiled from source and depend on packages like torch don't have to be precompiled for MLOPs.