Upload 52 files

.gitattributes

@@ -56,3 +56,27 @@ videos/ShadowHandDoorOpenInward/rl-video-episode-6.gif filter=lfs diff=lfs merge
 videos/ShadowHandDoorOpenInward/rl-video-episode-7.gif filter=lfs diff=lfs merge=lfs -text
 videos/ShadowHandDoorOpenInward/rl-video-episode-8.gif filter=lfs diff=lfs merge=lfs -text
 videos/ShadowHandDoorOpenInward/rl-video-episode-9.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/rl-video-episode-0.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/rl-video-episode-1.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/rl-video-episode-2.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/rl-video-episode-3.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-0.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-1.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-2.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-3.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-4.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-5.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-6.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-7.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-8.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandBlockStack/rl-video-episode-9.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-0.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-1.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-2.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-3.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-4.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-5.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-6.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-7.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-8.gif filter=lfs diff=lfs merge=lfs -text
+robotinder-data/ShadowHandDoorOpenInward/rl-video-episode-9.gif filter=lfs diff=lfs merge=lfs -text

robotinder-data/.gitignore

@@ -0,0 +1 @@
+rl*.mp4

robotinder-data/data_info.json

@@ -0,0 +1 @@
+{"idx": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20], "env": ["ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandBlockStack", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward", "ShadowHandDoorOpenInward"], "name": ["rl-video-episode-1.gif", "rl-video-episode-9.gif", "rl-video-episode-7.gif", "rl-video-episode-2.gif", "rl-video-episode-3.gif", "rl-video-episode-6.gif", "rl-video-episode-0.gif", "rl-video-episode-4.gif", "rl-video-episode-5.gif", "rl-video-episode-8.gif", "rl-video-episode-1.gif", "rl-video-episode-9.gif", "rl-video-episode-7.gif", "rl-video-episode-2.gif", "rl-video-episode-3.gif", "rl-video-episode-6.gif", "rl-video-episode-0.gif", "rl-video-episode-4.gif", "rl-video-episode-5.gif", "rl-video-episode-8.gif"]}

robotinder-data/data_parser.py

@@ -0,0 +1,27 @@
+import json
+
+from os import listdir
+from os.path import isfile, join, isdir
+mypath = './'
+onlyfolders = [f for f in listdir(mypath) if isdir(join(mypath, f))]
+
+idx = 0
+data_info = {'idx': [], 'env': [], 'name': []}
+for folder in onlyfolders:
+    print(folder)
+    fs = [join(mypath, folder, f) for f in listdir(join(mypath, folder)) if isfile(join(mypath, folder, f))]
+    # print(fs)
+    for f in fs:
+        if f.endswith(".gif"):
+            idx += 1
+            data_info['idx'].append(idx)
+            data_info['env'].append(folder)
+            data_info['name'].append(f.split('/')[-1])
+
+with open('data_info.json', 'w') as f:
+    json.dump(data_info, f)
+    
+with open('data_info.json', 'r') as f:
+    data_info = json.load(f)
+print(data_info)
+    

robotinder-data/mp4togif.py

@@ -0,0 +1,17 @@
+from moviepy.editor import VideoFileClip
+
+from os import listdir
+from os.path import isfile, join, isdir
+mypath = './'
+onlyfolders = [f for f in listdir(mypath) if isdir(join(mypath, f))]
+
+for folder in onlyfolders:
+    print(folder)
+    fs = [join(mypath, folder, f) for f in listdir(join(mypath, folder)) if isfile(join(mypath, folder, f))]
+    print(fs)
+    for f in fs:
+        if f.endswith(".mp4"):
+            clip = VideoFileClip(f)
+            clip = clip.resize(height=512)
+            clip = clip.set_fps(10)
+            clip.write_gif(f.replace('.mp4', '.gif'), program='ffmpeg')