• Luya Gao's avatar
    Skeleton of ShapeNetCore class · 9d279ba5
    Luya Gao authored
    Summary: Skeleton of ShapeNetCore class that loads ShapeNet v1 from a given directory to a Dataset object. Overrides _init_, _len_, and _getitem_ from torch.utils.data.Dataset. Currently getitem returns verts, faces and id_str, where id_str is a concatenation of synset_id and obj_id. Planning on adding support for loading ShapeNet v2, retrieving textures and returning wordnet synsets (not just ids) in next diffs.
    
    Reviewed By: nikhilaravi
    
    Differential Revision: D21986222
    
    fbshipit-source-id: c2c515303f1898b6c495b52cb53c74d691585326
    9d279ba5
test_shapenet_core.py 2.33 KB