update link

kelvin34501 · Jun 25, 2022 · ae766db · ae766db
1 parent 67f9441
commit ae766db
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/oikit/oi_shape/README.md b/oikit/oi_shape/README.md
@@ -4,8 +4,8 @@ To visualize the OakInk-Shape part, please follow the instructions below.
 
 Step 1:
 
-* Download [metaV2](xxx), [oakink_shape_v2](xxx), and the two object files (including [real](xxxx) and [virtual](xxxx)]). Then, unzip the files and put them into the `data` directory.
-* Download [mano](xxxx) following the official instructions. Then, put it into the `assets` directory.
+* Download [metaV2](https://www.dropbox.com/s/uh7o95lx0qbp05r/metaV2.zip?dl=0), [oakink_shape_v2](https://www.dropbox.com/s/swqintyk2g64bed/oakink_shape_v2.zip?dl=0), and the two object files (including [real](https://www.dropbox.com/s/sq38cjn4finrt2q/OakInkObjectsV2.zip?dl=0) and [virtual](https://www.dropbox.com/s/1zgy6zjk33s5q6y/OakInkVirtualObjectsV2.zip?dl=0)]). Then, unzip the files and put them into the `data` directory.
+* Download [mano](https://mano.is.tue.mpg.de) following the official instructions. Then, put it into the `assets` directory.
 
 Step 2:
 

diff --git a/oikit/oi_shape/oi_shape.py b/oikit/oi_shape/oi_shape.py
@@ -57,7 +57,8 @@ def __init__(
             virtual_matcher = re.compile(rf"({cat}/(.{{6}})/.{{10}})/(.{{6}})/hand_param\.pkl$")
             path = os.path.join(self.data_root_dir, cat)
             category_begin_idx.append(len(grasp_list))
-            for cur, _, files in os.walk(path, followlinks=False):
+            for cur, dirs, files in os.walk(path, followlinks=False):
+                dirs.sort()
                 for f in files:
                     re_match = virtual_matcher.findall(os.path.join(cur, f))
                     is_virtual = len(re_match) > 0

diff --git a/oikit/oi_shape/oi_utils.py b/oikit/oi_shape/oi_utils.py
@@ -82,10 +82,10 @@ def get_obj_path(oid, data_path, meta_path, use_downsample=True, key="align"):
     virtual_meta = json.load(open(os.path.join(meta_path, "virtual_object_id.json"), "r"))
     if oid in real_meta:
         obj_name = real_meta[oid]["name"]
-        obj_path = os.path.join(data_path, "yodaObjectsV2")
+        obj_path = os.path.join(data_path, "OakInkObjectsV2")
     else:
         obj_name = virtual_meta[oid]["name"]
-        obj_path = os.path.join(data_path, "yodaVirtualObjectsV2")
+        obj_path = os.path.join(data_path, "OakInkVirtualObjectsV2")
     obj_mesh_path = list(
         glob.glob(os.path.join(obj_path, obj_name, obj_suffix_path, "*.obj"))
         + glob.glob(os.path.join(obj_path, obj_name, obj_suffix_path, "*.ply"))