ThibaultGROUEIX / AtlasNet

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.
http://imagine.enpc.fr/~groueixt/atlasnet/
MIT License
668 stars 117 forks source link

Evaluate RGB image with pretrained model #29

Closed Itamare1982 closed 5 years ago

Itamare1982 commented 5 years ago

Hi Iam actually try to evaluate SVR Atlas pretrained model on RGB image(chair), my parameters are really similar to the demo and i got wird result (by view in chrom 3D viewer ). i used the demo grid generation. when i run your demo plane.jpg im my network i got good results in the 3Dviewer . Demo plain wird_pic_1 wird_pic_2 wird_pic_3 can you please direct my how to evaluate RGB image?

ThibaultGROUEIX commented 5 years ago

hi @Itamare1982, could you share your input images please? i think it's probably a background color issue. make sure it's the same as the demo image cheers Thibault

Itamare1982 commented 5 years ago

it seems like the problem is with the background - but how can i fix it? to be as the plain demo background? you can find attached my image chair

ThibaultGROUEIX commented 5 years ago

thanks for sharing. You need the background to be all black or all white (don't remember which). there are two options : (1) retrain the models with data augmentation to deal with more diverse backgrounds (2) fix this particular image. i think it can be easily done on Photoshop

Itamare1982 commented 5 years ago

Hi , i try to change the background ( as in the attached image) with a simple OpenCV script and i got the attached output which is also not as excpected. any suggestion ? 3D_output chair_chess_board

Itamare1982 commented 5 years ago

Black background doesnt help either chair_black_background

Itamare1982 commented 5 years ago

Sorry i missed your last comment about the Black/White background

Itamare1982 commented 5 years ago

Hi - white background doesnt solve the issue also and i am continue getting bad 3D object do you have any other suggestions?

chair_white_background white_background_output

ThibaultGROUEIX commented 5 years ago

I removed the background with the instant alpha of preview in mac then gave it to demo. It seems not great but reasonable. 4412 65286632-f2d9d080-db48-11e9-8195-4b9456f57fdf

Itamare1982 commented 5 years ago

OK ,thank you very much for the help and for the sharing, i do get better result without background but it still not as expected , i will continue to improve it and get better results.

Thank you for the productive suggestions