Git Product home page Git Product logo

rasmusrpaulsen / deep-mvlm Goto Github PK

View Code? Open in Web Editor NEW
119.0 8.0 24.0 11.85 MB

A tool for precisely placing 3D landmarks on 3D facial scans based on the paper "Multi-view Consensus CNN for 3D Facial Landmark Placement"

Home Page: http://shapeml.compute.dtu.dk/

License: MIT License

Python 100.00%
landmark-detection 3d-models deep-learning neural-network hourglass-network pytorch-implementation landmark-estimation 3d-landmarks facial-landmarks 3d-face-recognition

deep-mvlm's People

Contributors

rasmusrpaulsen avatar

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

deep-mvlm's Issues

AttributeError: 'NoneType' object has no attribute 'GetMapper'

Hi, thank you for your wonderful work, I have learned a lot from your code. But could you have me with this question?
I want to prepare rendered data using my raw data (.wrl, .bmp, and the landmark .txt files). But when run the preparedata.py file, I got the following problems. Unfortunately, I can not solve it. Do you know anything about it? I guess there is some format difference between your data and mine. If it is possible, could you share a set of example data for training? Maybe the data from DTU dataset because the BU_3DFE dataset is not available to download freely.

Preparing BU-3DFE data
Read 10 file ids
Processing 7 file ids for training
Processing F0001\F0001_AN02WH
F0001\F0001_AN02WH is locked - skipping
Processing F0001\F0001_AN03WH
F0001\F0001_AN03WH is locked - skipping
Processing F0001\F0001_AN04WH
F0001\F0001_AN04WH is locked - skipping
Processing F0001\F0001_DI01WH
Rendering F0001\F0001_DI01WH
Error near line 2: parse error
Traceback (most recent call last):
File "D:/07_AutomaticLandmarking/Paulsen_DeepMVLM/preparedata.py", line 349, in
main(cfg_global)
File "D:/07_AutomaticLandmarking/Paulsen_DeepMVLM/preparedata.py", line 334, in main
prepare_bu_3dfe_data(config)
File "D:/07_AutomaticLandmarking/Paulsen_DeepMVLM/preparedata.py", line 327, in prepare_bu_3dfe_data
process_file_bu_3dfe(config, base_name, output_dir)
File "D:/07_AutomaticLandmarking/Paulsen_DeepMVLM/preparedata.py", line 97, in process_file_bu_3dfe
pd = vrmlin.GetRenderer().GetActors().GetLastActor().GetMapper().GetInput()
AttributeError: 'NoneType' object has no attribute 'GetMapper'

The landmark placing on the MRI head model

Hi everybody,

I have two questions:

  1. Can I place the facial landmarks on the following MRI head model (STL or PLY or OBJ) with Deep-MVLM? It has 500.000 vertices and 800.000 faces. However, there is no information other than the STL file.

image

  1. By navigating around a head prototype with the depth camera, I obtained 10 RGB-D data. Can I create the head model (3d head.stl) using this code?

Thank you for this sharing

Is it able to support the xyz format file?

First of all, thanks to share your great 3D face landmark project.
I am trying to recognize a face from a .xyz cloud point data file, however you support OBJ/WRL/VTK/PLY only, because I am using a TOF sensor and able to capture xyz data only.
So would you please help me to make sure the xyz format able to support in your project.

Thanks & Best Regards
Sui

DTU-3D dataset

Hi, Prof .Rasmus R. Paulsen,
Is the DTU-3D dataset public available?
I try to train the dataset with SOTA network to improve the precision.

Getting wrong landamrks with each config file

Hi!
Thank you for sharing your great work with the community.
I would like to ask wich config file should I use for the model below, I tried geometry and depth, but without sucess. Is it something wrong with my mesh? Thank you.

image

Implementation of 3D landmark annotation

Hi,
Can you share the implementation of 3D landmark annotation of DTU3D ?
The orignal annotation of BU-3DFE dataset seems unsatisfactory and does not provide the annotation of the nose tip that your annotation does.

Any help would be appreciate !

Regarding custom dataset

Hi. I have a query regarding the implementation using custom dataset.
I have .obj mesh file (example screenshot is attached)
example

I wanted to test the model on it but I get the message "Not enough valid view lines for landmark" for all the landmarks.

I wanted to ask if you can guide on how to train the model on custom dataset and how to annotate.

White Image View

Hello, may I ask, if I test my own data, I find that first of all, my .obj file has vn, while the test data does not, will this affect? I also find that when I import the data into the model, the generated views are all white images. What is the reason? Thank you very much for your answer

NotImplementedError: cannot instantiate 'WindowsPath' on your system

When I try to run this code on an AWS SageMaker notebook instance:
python predict.py --c configs/DTU3D-RGB.json --n assets/testmeshA.obj

I run in to the following error:
Processing assets/testmeshA.obj
Initialising model
Loading checkpoint
Getting device
Warning: There's no GPU available on this machine,prediction will be performed on CPU.
Loading checkpoint: https://shapeml.compute.dtu.dk/Deep-MVLM/models/MVLMModel_DTU3D_RGB_07092019-c1cc3d59.pth
Traceback (most recent call last):
File "predict.py", line 70, in
main(global_config)
File "predict.py", line 51, in main
process_one_file(config, name)
File "predict.py", line 12, in process_one_file
dm = deepmvlm.DeepMVLM(config)
File "/home/ec2-user/SageMaker/Deep-MVLM/deepmvlm/api.py", line 37, in init
self.device, self.model = self._get_device_and_load_model_from_url()
File "/home/ec2-user/SageMaker/Deep-MVLM/deepmvlm/api.py", line 75, in _get_device_and_load_model_from_url
checkpoint = load_url(check_point_name, model_dir, map_location=device)
File "/home/ec2-user/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/hub.py", line 506, in load_state_dict_from_url
return torch.load(cached_file, map_location=map_location)
File "/home/ec2-user/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/serialization.py", line 529, in load
return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args)
File "/home/ec2-user/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/serialization.py", line 702, in _legacy_load
result = unpickler.load()
File "/home/ec2-user/anaconda3/envs/pytorch_p36/lib/python3.6/pathlib.py", line 1002, in new
% (cls.name,))
NotImplementedError: cannot instantiate 'WindowsPath' on your system

How to get Deep-MVLM style shading on mesh?

image

On the left is the my mesh put through the Deep-MVLM algorithm and visualised with its built in functions. On the right is when I open the same mesh in mesh lab. How do I get the lighting/shading similar to how you all did it? It brings out the contours much better and it would help to know exactly how you did it.

Thanks.

DTU3D-depth on the man_bust.obj example

Hello,
I have an issue when I am running the explained command : python predict.py --c configs\DTU3D-depth.json --n man_bust.obj
image

The landmarks does not calculate themself well and don't appear on the render window.
image

I did not change any part of the code.
What can I do ?

Thanks

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.