Upload app.py

app.py

@@ -127,11 +127,30 @@ uids = sorted(uids)
 # uids = uids[:809]
 # uids = uids[809:1619]
 # uids = uids[1619:]
-uids = uids[:2]
+
+dict = {
+        "cracker_box": "dbbfa480a2e4476c8b0ca91667087f44",
+        "packet": "82ab1bb22a22407e8ebd86cae46dc9d3",
+        "mug": "a831091dabdb40598ed1681e2aa35058",
+        "chips_can": "6848fd0ec6df45bcb53fc0e47b51e844",
+        "baseball": "4799dc6cb5f046bd88609381020feab1",
+        "paper_towel": "d020130cdc0b4937a062f8d744d707c6",
+        "banana": "1de9ac4b1a4e46279c271747becbf0eb",
+        "bowl": "9fa4ca710fa9458aaeb9554c80583c76",
+        "pear": "fa74a39529f5462e978b4f5dbc0b8add",
+        "sponge": "939ca8d1e0c2424ba28db9afd19e7e53",
+        "tennis_ball": "05eb4a68e56249c5b3db4ea937a57b6e",
+        "plastic_bag": "686c497301da4fa5b6927b79340f473f"
+    }
+new_uids = []
+for dict_key_one in dict.keys():
+    new_uids_one = dict[dict_key_one]
+    if new_uids_one not in uids:
+        new_uids.append(new_uids_one)
 
 
 objects = objaverse.load_objects(
-    uids=uids,
+    uids=new_uids,
     download_processes=processes
 )
 
@@ -140,13 +159,12 @@ repo = Repository(
 )
 
 
-for objaid in tqdm(uids):
+for objaid in tqdm(new_uids):
     objamodel = objaverse.load_objects([objaid])[objaid]
     try:
         pc = trimesh_to_pc(trimesh.load(objamodel))
         pc = random_sampling(pc, num_samples=5000)
         object_id_tmp_dict[objaid] = pc
-        print(pc.shape)
     except:
         continue