kopia lustrzana https://github.com/lzzcd001/MeshDiffusion
33 wiersze
1.0 KiB
Python
33 wiersze
1.0 KiB
Python
|
# Copyright (c) 2020-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
||
|
|
#
|
||
|
|
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
|
||
|
|
# property and proprietary rights in and to this material, related
|
||
|
|
# documentation and any modifications thereto. Any use, reproduction,
|
||
|
|
# disclosure or distribution of this material and related documentation
|
||
|
|
# without an express license agreement from NVIDIA CORPORATION or
|
||
|
|
# its affiliates is strictly prohibited.
|
||
|
|
|
||
|
|
import numpy as np
|
||
|
|
import torch
|
||
|
|
|
||
|
|
import os
|
||
|
|
import json
|
||
|
|
|
||
|
|
###############################################################################
|
||
|
|
# ShapeNet Dataset to get meshes
|
||
|
|
###############################################################################
|
||
|
|
|
||
|
|
class ShapeNetDataset(object):
|
||
|
|
def __init__(self, shapenet_json):
|
||
|
|
with open(shapenet_json, 'r') as f:
|
||
|
|
self.mesh_list = json.load(f)
|
||
|
|
|
||
|
|
print(f"len all data: {len(self.mesh_list)}")
|
||
|
|
|
||
|
|
def __len__(self):
|
||
|
|
return len(self.mesh_list)
|
||
|
|
|
||
|
|
|
||
|
|
def __getitem__(self, idx):
|
||
|
|
return self.mesh_list[idx]
|