Upload 2 files

app.py

@@ -11,7 +11,7 @@ import json
 from tqdm import tqdm
 
 DATASET_REPO_URL = "https://huggingface.co/datasets/Gary3410/object_tmp"
-DATA_FILENAME = "object_tmp_4.pkl"
+DATA_FILENAME = "object_tmp_22_1.pkl"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
 
 
@@ -110,15 +110,18 @@ uids = []
 object_id_tmp_dict = {}
 
 # 解析json文件
-with open('object_id.json','r') as file:
+with open('object_22.json','r') as file:
     str = file.read()
     obj_data = json.loads(str)
 
 for ints_key in obj_data.keys():
     ints_dict_one = obj_data[ints_key]
-    ints_id_dict = ints_dict_one["obj_id"]
-    for ints_one in ints_id_dict.keys():
-        uid_one = ints_id_dict[ints_one]
+    # ints_id_dict = ints_dict_one["obj_id"]
+    # for ints_one in ints_id_dict.keys():
+    #     uid_one = ints_id_dict[ints_one]
+    #     uids.append(uid_one)
+    for ints_one in ints_dict_one.keys():
+        uid_one = ints_dict_one[ints_one]
         uids.append(uid_one)
 
 uids = list(set(uids))
@@ -128,29 +131,29 @@ uids = sorted(uids)
 # uids = uids[809:1619]
 # uids = uids[1619:]
 
-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)
+# 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=new_uids,
+    uids=uids,
     download_processes=processes
 )
 
@@ -159,7 +162,7 @@ repo = Repository(
 )
 
 
-for objaid in tqdm(new_uids):
+for objaid in tqdm(uids):
     objamodel = objaverse.load_objects([objaid])[objaid]
     try:
         pc = trimesh_to_pc(trimesh.load(objamodel))