Python TriangleMesh.from_offの例

コード例 #1

ファイル: modelnet.py プロジェクト: stjordanis/kaolin-1

    def __getitem__(self, index):
        """Returns the item at index idx. """
        category = torch.tensor(self.cat_idxs[index],
                                dtype=torch.long,
                                device=self.device)
        data = TriangleMesh.from_off(self.filepaths[index])
        data.to(self.device)
        if self.transform:
            data = self.transform(data)

        return data, category

コード例 #2

def load_all():
    base = Path('data/ModelNet40')
    all_models = list(base.glob('**/*.off'))
    bad = []
    for model in all_models:
        try:
            data = TriangleMesh.from_off(model)
            samp = data.sample(16)
        except:
            print(model)
            bad.append(model)
            raise
    print(bad)

コード例 #3

 def _get_data(self, index):
     data = TriangleMesh.from_off(self.filepaths[index])
     return data