🚀 [RofuncRL] Add two more dexterous task pre-trained models to ModelZoo

Skylark0924 · Dec 30, 2023 · 935b54b · 935b54b
1 parent 45cfb11
commit 935b54b
Show file tree

Hide file tree

Showing 25 changed files with 52 additions and 47 deletions.
diff --git a/README.md b/README.md
diff --git a/doc/img/AntRofuncRLPPO.mp4 b/doc/img/AntRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandBlockStackRofuncRLPPO.mp4 b/doc/img/BiShadowHandBlockStackRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandBottleCapRofuncRLPPO.mp4 b/doc/img/BiShadowHandBottleCapRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandCatchAbreastRofuncRLPPO.mp4 b/doc/img/BiShadowHandCatchAbreastRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandCatchOver2UnderarmRofuncRLPPO.mp4 b/doc/img/BiShadowHandCatchOver2UnderarmRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandCatchUnderarmRofuncRLPPO.mp4 b/doc/img/BiShadowHandCatchUnderarmRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandDoorCloseInwardRofuncRLPPO.mp4 b/doc/img/BiShadowHandDoorCloseInwardRofuncRLPPO.mp4
diff --git a/...noidASELocationSwordShieldRofuncRLPPO.mp4 → ...ShadowHandDoorCloseOutwardRofuncRLPPO.gif b/...noidASELocationSwordShieldRofuncRLPPO.mp4 → ...ShadowHandDoorCloseOutwardRofuncRLPPO.gif
diff --git a/doc/img/BiShadowHandDoorOpenInwardRofuncRLPPO.gif b/doc/img/BiShadowHandDoorOpenInwardRofuncRLPPO.gif
diff --git a/doc/img/BiShadowHandDoorOpenOutwardRofuncRLPPO.mp4 b/doc/img/BiShadowHandDoorOpenOutwardRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandGraspAndPlaceRofuncRLPPO.mp4 b/doc/img/BiShadowHandGraspAndPlaceRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandLiftUnderarmRofuncRLPPO.mp4 b/doc/img/BiShadowHandLiftUnderarmRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandOverRofuncRLPPO.mp4 b/doc/img/BiShadowHandOverRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandPenRofuncRLPPO.mp4 b/doc/img/BiShadowHandPenRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandPushBlockRofuncRLPPO.mp4 b/doc/img/BiShadowHandPushBlockRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandReOrientationRofuncRLPPO.mp4 b/doc/img/BiShadowHandReOrientationRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandScissorsRofuncRLPPO.mp4 b/doc/img/BiShadowHandScissorsRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandSwingCupRofuncRLPPO.mp4 b/doc/img/BiShadowHandSwingCupRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandSwitchRofuncRLPPO.mp4 b/doc/img/BiShadowHandSwitchRofuncRLPPO.mp4
diff --git a/doc/img/BiShadowHandTwoCatchUnderarmRofuncRLPPO.gif b/doc/img/BiShadowHandTwoCatchUnderarmRofuncRLPPO.gif
diff --git a/doc/img/FrankaCabinetRofuncRLPPO.mp4 b/doc/img/FrankaCabinetRofuncRLPPO.mp4
diff --git a/doc/img/HumanoidRofuncRLPPO.mp4 b/doc/img/HumanoidRofuncRLPPO.mp4
diff --git a/examples/learning_rl/README.rst b/examples/learning_rl/README.rst
@@ -238,7 +238,7 @@ The following are examples of reinforcement learning methods for robot learning.
              -
              - ✅
            * - `BiShadowHandDoorOpenInward <https://github.com/Skylark0924/Rofunc/blob/main/rofunc/learning/RofuncRL/tasks/isaacgymenv/hands/shadow_hand_door_open_inward.py>`_
-             - 
+             - .. image:: ../../../img/BiShadowHandDoorOpenInwardRofuncRLPPO.gif
              -
              - ✅
            * - `BiShadowHandDoorOpenOutward <https://github.com/Skylark0924/Rofunc/blob/main/rofunc/learning/RofuncRL/tasks/isaacgymenv/hands/shadow_hand_door_open_outward.py>`_
@@ -250,7 +250,7 @@ The following are examples of reinforcement learning methods for robot learning.
              - 
              - ✅
            * - `BiShadowHandDoorCloseOutward <https://github.com/Skylark0924/Rofunc/blob/main/rofunc/learning/RofuncRL/tasks/isaacgymenv/hands/shadow_hand_door_close_outward.py>`_
-             - 
+             - .. image:: ../../../img/BiShadowHandDoorCloseOutwardRofuncRLPPO.gif
              -
              - ✅
            * - `BiShadowHandGraspAndPlace <https://github.com/Skylark0924/Rofunc/blob/main/rofunc/learning/RofuncRL/tasks/isaacgymenv/hands/shadow_hand_grasp_and_place.py>`_
@@ -294,9 +294,9 @@ The following are examples of reinforcement learning methods for robot learning.
              -
              - ✅
            * - `BiShadowHandTwoCatchUnderarm <https://github.com/Skylark0924/Rofunc/blob/main/rofunc/learning/RofuncRL/tasks/isaacgymenv/hands/shadow_hand_two_catch_underarm.py>`_
-             - 
-             -
+             - .. image:: ../../../img/BiShadowHandTwoCatchUnderarmRofuncRLPPO.gif
              -
+             - ✅
 
 
 

diff --git a/rofunc/config/learning/model_zoo.json b/rofunc/config/learning/model_zoo.json
@@ -110,7 +110,7 @@
     "md5": ""
   },
   "BiShadowHandDoorOpenInwardRofuncRLPPO.pth": {
-    "url": "https://drive.google.com/uc?id=16Okt9cIAvTb8yVSheftbEozC6ZL4_oZA&export=download",
+    "url": "https://drive.google.com/uc?id=1fTguenc0p1PbUsM7ojhip1ZcFbytpQy_&export=download",
     "name": "BiShadowHandDoorOpenInwardRofuncRLPPO.pth",
     "md5": ""
   },
@@ -125,7 +125,7 @@
     "md5": ""
   },
   "BiShadowHandDoorCloseOutwardRofuncRLPPO.pth": {
-    "url": "https://drive.google.com/uc?id=1z0DjM7h2CikaldaC5Sm4p2MG4wYDrRc_&export=download",
+    "url": "https://drive.google.com/uc?id=10EU-kEgapbedoWiuH9UVGOjxGFO0T93u&export=download",
     "name": "BiShadowHandDoorCloseOutwardRofuncRLPPO.pth",
     "md5": ""
   },
@@ -174,6 +174,11 @@
     "name": "BiShadowHandSwitchRofuncRLPPO.pth",
     "md5": ""
   },
+  "BiShadowHandTwoCatchUnderarmRofuncRLPPO.pth": {
+    "url": "https://drive.google.com/uc?id=1TWXp60AKV-P48_JXvA64-Qi2oijriNAn&export=download",
+    "name": "BiShadowHandTwoCatchUnderarmRofuncRLPPO.pth",
+    "md5": ""
+  },
   "efficientsam_s_gpu.jit": {
     "url": "https://drive.google.com/uc?id=1lMCUFOLeQOlIpuaIA00MB174lL7ZiuoS&export=download",
     "name": "efficientsam_s_gpu.jit",