├── .gitignore ├── LICENSE ├── README.md ├── configs ├── car.yml ├── car_big.yml ├── cub.yml ├── cub_big.yml ├── custom.yml ├── custom_big.yml ├── horse.yml ├── horse_big.yml ├── moto.yml ├── moto_big.yml ├── p3d_car.yml ├── p3d_car_big.yml ├── sn │ ├── airplane.yml │ ├── bench.yml │ ├── cabinet.yml │ ├── car.yml │ ├── chair.yml │ ├── display.yml │ ├── lamp.yml │ ├── phone.yml │ ├── rifle.yml │ ├── sofa.yml │ ├── speaker.yml │ ├── table.yml │ └── vessel.yml └── sn_big │ ├── airplane.yml │ ├── bench.yml │ ├── cabinet.yml │ ├── car.yml │ ├── chair.yml │ ├── display.yml │ ├── lamp.yml │ ├── phone.yml │ ├── rifle.yml │ ├── sofa.yml │ ├── speaker.yml │ ├── table.yml │ └── vessel.yml ├── demo ├── 00.jpg ├── 01.jpg ├── 02.jpg ├── 03.jpg ├── 04.jpg ├── 05.jpg └── 06.jpg ├── environment.yml ├── media ├── example_car.png ├── example_rec.gif ├── shapenet.gif ├── teaser.gif └── teaser.mp4 ├── scripts ├── download_data.sh ├── download_model.sh ├── kp_eval.sh ├── pipeline.sh ├── reconstruct.sh └── set_params.sh └── src ├── __init__.py ├── dataset ├── __init__.py ├── comp_cars.py ├── cub_200.py ├── folder.py ├── lsun.py ├── p3d_car.py ├── shapenet.py └── torch_transforms.py ├── kp_eval.py ├── model ├── __init__.py ├── encoder.py ├── field.py ├── generator.py ├── loss.py ├── pytorch3d_monkey.py ├── renderer.py ├── tools.py └── unicorn.py ├── optimizer.py ├── reconstruct.py ├── scheduler.py ├── trainer.py └── utils ├── __init__.py ├── chamfer.py ├── icp.py ├── image.py ├── logger.py ├── mesh.py ├── metrics.py ├── path.py ├── plot.py └── pytorch.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/README.md -------------------------------------------------------------------------------- /configs/car.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/car.yml -------------------------------------------------------------------------------- /configs/car_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/car_big.yml -------------------------------------------------------------------------------- /configs/cub.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/cub.yml -------------------------------------------------------------------------------- /configs/cub_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/cub_big.yml -------------------------------------------------------------------------------- /configs/custom.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/custom.yml -------------------------------------------------------------------------------- /configs/custom_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/custom_big.yml -------------------------------------------------------------------------------- /configs/horse.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/horse.yml -------------------------------------------------------------------------------- /configs/horse_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/horse_big.yml -------------------------------------------------------------------------------- /configs/moto.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/moto.yml -------------------------------------------------------------------------------- /configs/moto_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/moto_big.yml -------------------------------------------------------------------------------- /configs/p3d_car.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/p3d_car.yml -------------------------------------------------------------------------------- /configs/p3d_car_big.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/p3d_car_big.yml -------------------------------------------------------------------------------- /configs/sn/airplane.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/airplane.yml -------------------------------------------------------------------------------- /configs/sn/bench.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/bench.yml -------------------------------------------------------------------------------- /configs/sn/cabinet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/cabinet.yml -------------------------------------------------------------------------------- /configs/sn/car.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/car.yml -------------------------------------------------------------------------------- /configs/sn/chair.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/chair.yml -------------------------------------------------------------------------------- /configs/sn/display.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/display.yml -------------------------------------------------------------------------------- /configs/sn/lamp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/lamp.yml -------------------------------------------------------------------------------- /configs/sn/phone.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/phone.yml -------------------------------------------------------------------------------- /configs/sn/rifle.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/rifle.yml -------------------------------------------------------------------------------- /configs/sn/sofa.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/sofa.yml -------------------------------------------------------------------------------- /configs/sn/speaker.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/speaker.yml -------------------------------------------------------------------------------- /configs/sn/table.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/table.yml -------------------------------------------------------------------------------- /configs/sn/vessel.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn/vessel.yml -------------------------------------------------------------------------------- /configs/sn_big/airplane.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/airplane.yml -------------------------------------------------------------------------------- /configs/sn_big/bench.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/bench.yml -------------------------------------------------------------------------------- /configs/sn_big/cabinet.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/cabinet.yml -------------------------------------------------------------------------------- /configs/sn_big/car.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/car.yml -------------------------------------------------------------------------------- /configs/sn_big/chair.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/chair.yml -------------------------------------------------------------------------------- /configs/sn_big/display.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/display.yml -------------------------------------------------------------------------------- /configs/sn_big/lamp.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/lamp.yml -------------------------------------------------------------------------------- /configs/sn_big/phone.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/phone.yml -------------------------------------------------------------------------------- /configs/sn_big/rifle.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/rifle.yml -------------------------------------------------------------------------------- /configs/sn_big/sofa.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/sofa.yml -------------------------------------------------------------------------------- /configs/sn_big/speaker.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/speaker.yml -------------------------------------------------------------------------------- /configs/sn_big/table.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/table.yml -------------------------------------------------------------------------------- /configs/sn_big/vessel.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/configs/sn_big/vessel.yml -------------------------------------------------------------------------------- /demo/00.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/00.jpg -------------------------------------------------------------------------------- /demo/01.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/01.jpg -------------------------------------------------------------------------------- /demo/02.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/02.jpg -------------------------------------------------------------------------------- /demo/03.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/03.jpg -------------------------------------------------------------------------------- /demo/04.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/04.jpg -------------------------------------------------------------------------------- /demo/05.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/05.jpg -------------------------------------------------------------------------------- /demo/06.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/demo/06.jpg -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/environment.yml -------------------------------------------------------------------------------- /media/example_car.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/media/example_car.png -------------------------------------------------------------------------------- /media/example_rec.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/media/example_rec.gif -------------------------------------------------------------------------------- /media/shapenet.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/media/shapenet.gif -------------------------------------------------------------------------------- /media/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/media/teaser.gif -------------------------------------------------------------------------------- /media/teaser.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/media/teaser.mp4 -------------------------------------------------------------------------------- /scripts/download_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/scripts/download_data.sh -------------------------------------------------------------------------------- /scripts/download_model.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/scripts/download_model.sh -------------------------------------------------------------------------------- /scripts/kp_eval.sh: -------------------------------------------------------------------------------- 1 | #!/bin/bash 2 | set -e 3 | CUDA_VISIBLE_DEVICES=$cuda python src/kp_eval.py --tag $tag 4 | -------------------------------------------------------------------------------- /scripts/pipeline.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/scripts/pipeline.sh -------------------------------------------------------------------------------- /scripts/reconstruct.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/scripts/reconstruct.sh -------------------------------------------------------------------------------- /scripts/set_params.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/scripts/set_params.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /src/dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/__init__.py -------------------------------------------------------------------------------- /src/dataset/comp_cars.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/comp_cars.py -------------------------------------------------------------------------------- /src/dataset/cub_200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/cub_200.py -------------------------------------------------------------------------------- /src/dataset/folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/folder.py -------------------------------------------------------------------------------- /src/dataset/lsun.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/lsun.py -------------------------------------------------------------------------------- /src/dataset/p3d_car.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/p3d_car.py -------------------------------------------------------------------------------- /src/dataset/shapenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/shapenet.py -------------------------------------------------------------------------------- /src/dataset/torch_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/dataset/torch_transforms.py -------------------------------------------------------------------------------- /src/kp_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/kp_eval.py -------------------------------------------------------------------------------- /src/model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/__init__.py -------------------------------------------------------------------------------- /src/model/encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/encoder.py -------------------------------------------------------------------------------- /src/model/field.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/field.py -------------------------------------------------------------------------------- /src/model/generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/generator.py -------------------------------------------------------------------------------- /src/model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/loss.py -------------------------------------------------------------------------------- /src/model/pytorch3d_monkey.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/pytorch3d_monkey.py -------------------------------------------------------------------------------- /src/model/renderer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/renderer.py -------------------------------------------------------------------------------- /src/model/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/tools.py -------------------------------------------------------------------------------- /src/model/unicorn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/model/unicorn.py -------------------------------------------------------------------------------- /src/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/optimizer.py -------------------------------------------------------------------------------- /src/reconstruct.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/reconstruct.py -------------------------------------------------------------------------------- /src/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/scheduler.py -------------------------------------------------------------------------------- /src/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/trainer.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/chamfer.py -------------------------------------------------------------------------------- /src/utils/icp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/icp.py -------------------------------------------------------------------------------- /src/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/image.py -------------------------------------------------------------------------------- /src/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/logger.py -------------------------------------------------------------------------------- /src/utils/mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/mesh.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/path.py -------------------------------------------------------------------------------- /src/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/plot.py -------------------------------------------------------------------------------- /src/utils/pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/monniert/unicorn/HEAD/src/utils/pytorch.py --------------------------------------------------------------------------------