Skip to content

bananananacat/Generation-of-3D-Objects

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 

Repository files navigation

#Big readme update soon! At this point, the paper with the code has been submitted to a computer vision conference.

Installing Requarements

foo@bar:~$ pip3 install -r requirements.txt

Installing kaolin lib:

(also here is documentation)

foo@bar:~$ pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu121
foo@bar:~$ pip install kaolin==0.15.0 -f https://nvidia-kaolin.s3.us-east-2.amazonaws.com/torch-2.1.1_cu121.html

PreCERMIT model

This model generates point cloud with 1024 points from given arbitrary number of photos/renders from different views.

directory with model with all scripts: https://github.com/bananananacat/Generation-of-3D-Objects/tree/main/model/models/PreCERMIT

CERMIT model

This model generates point cloud with N*k points from given point cloud with N points(quality enhansment).

directory with model with all scripts: https://github.com/bananananacat/Generation-of-3D-Objects/tree/main/model/models/CERMIT

Losses && Metrics

update soon

Dataset

We created our own dataset and used ShapeNet dataset. There are some scripts - render functions, functions to work with camera and lights, functions to create dataset, etc.

About

Generation of point clouds

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published