Gary3410 commited on
Commit
ed99f8a
1 Parent(s): 4ed50e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -11,7 +11,7 @@ import json
11
  from tqdm import tqdm
12
 
13
  DATASET_REPO_URL = "https://huggingface.co/datasets/Gary3410/object_tmp"
14
- DATA_FILENAME = "object_tmp_3.pkl"
15
  DATA_FILE = os.path.join("data", DATA_FILENAME)
16
 
17
 
@@ -126,7 +126,8 @@ uids = sorted(uids)
126
 
127
  # uids = uids[:809]
128
  # uids = uids[809:1619]
129
- uids = uids[1619:]
 
130
 
131
 
132
  objects = objaverse.load_objects(
@@ -145,6 +146,7 @@ for objaid in tqdm(uids):
145
  pc = trimesh_to_pc(trimesh.load(objamodel))
146
  pc = random_sampling(pc, num_samples=5000)
147
  object_id_tmp_dict[objaid] = pc
 
148
  except:
149
  continue
150
 
 
11
  from tqdm import tqdm
12
 
13
  DATASET_REPO_URL = "https://huggingface.co/datasets/Gary3410/object_tmp"
14
+ DATA_FILENAME = "object_tmp_4.pkl"
15
  DATA_FILE = os.path.join("data", DATA_FILENAME)
16
 
17
 
 
126
 
127
  # uids = uids[:809]
128
  # uids = uids[809:1619]
129
+ # uids = uids[1619:]
130
+ uids = uids[:2]
131
 
132
 
133
  objects = objaverse.load_objects(
 
146
  pc = trimesh_to_pc(trimesh.load(objamodel))
147
  pc = random_sampling(pc, num_samples=5000)
148
  object_id_tmp_dict[objaid] = pc
149
+ print(pc.shape)
150
  except:
151
  continue
152