• Nikhila Ravi's avatar
    Update load obj and compare with SoftRas · c9267ab7
    Nikhila Ravi authored
    Summary:
    Updated the load obj function to support creating of a per face texture map using the information in an .mtl file. Uses the approach from in SoftRasterizer.
    
    Currently I have ported in the SoftRasterizer code but this is only to help with comparison and will  be deleted before landing. The ShapeNet Test data will also be deleted.
    
    Here is the [Design doc](https://docs.google.com/document/d/1AUcLP4QwVSqlfLAUfbjM9ic5vYn9P54Ha8QbcVXW2eI/edit?usp=sharing).
    
    ## Added
    - texture atlas creation functions in PyTorch based on the SoftRas cuda implementation
    - tests to compare SoftRas vs PyTorch3D implementation to verify it matches (using real shapenet data with meshes consisting of multiple textures)
    - benchmarks tests
    
    ## Remaining todo:
    - add more tests for obj io to test the new functions and the two texturing options
    - replace the shapenet data with the output from SoftRas saved as a file.
    
    # MAIN FILES TO REVIEW
    
    - `obj_io.py`
    - `test_obj_io.py` [still some tests to be added but have comparisons with SoftRas for now]
    
    The reference SoftRas implementations are in `softras_load_obj.py` and `load_textures.cu`.
    
    Reviewed By: gkioxari
    
    Differential Revision: D20754859
    
    fbshipit-source-id: 42ace9dfb73f26e29d800c763f56d5b66c60c5e2
    c9267ab7
bm_mesh_io.py 2.37 KB