42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
# Copyright (c) 2022, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
#
|
|
# NVIDIA CORPORATION & AFFILIATES and its licensors retain all intellectual property
|
|
# and proprietary rights in and to this software, related documentation
|
|
# and any modifications thereto. Any use, reproduction, disclosure or
|
|
# distribution of this software and related documentation without an express
|
|
# license agreement from NVIDIA CORPORATION & AFFILIATES is strictly prohibited.
|
|
import os
|
|
|
|
|
|
def get_path(dataname=None):
|
|
dataset_path = {}
|
|
dataset_path['pointflow'] = [
|
|
'./data/ShapeNetCore.v2.PC15k/'
|
|
|
|
]
|
|
dataset_path['clip_forge_image'] = [
|
|
'./data/shapenet_render/'
|
|
]
|
|
|
|
if dataname is None:
|
|
return dataset_path
|
|
else:
|
|
assert(
|
|
dataname in dataset_path), f'not found {dataname}, only: {list(dataset_path.keys())}'
|
|
for p in dataset_path[dataname]:
|
|
print(f'searching: {dataname}, get: {p}')
|
|
if os.path.exists(p):
|
|
return p
|
|
ValueError(
|
|
f'all path not found for {dataname}, please double check: {dataset_path[dataname]}; or edit the datasets/data_path.py ')
|
|
|
|
|
|
def get_cache_path():
|
|
cache_list = ['/workspace/data_cache_local/data_stat/',
|
|
'/workspace/data_cache/data_stat/']
|
|
for p in cache_list:
|
|
if os.path.exists(p):
|
|
return p
|
|
ValueError(
|
|
f'all path not found for {cache_list}, please double check: or edit the datasets/data_path.py ')
|