tangger/openpi
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update convert aloha to save images in RGB (#304)

Browse Source
This commit is contained in:
Michael Equi
2025-02-14 11:15:58 -08:00
committed by GitHub
parent 90b87cc42c 8d288e4b41
commit cd82848a99
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/aloha_real/convert_aloha_data_to_lerobot.py
View File

@@ -155,7 +155,7 @@ def load_raw_images_per_camera(ep: h5py.File, cameras: list[str]) -> dict[str, n
# load one compressed image after the other in RAM and uncompress # load one compressed image after the other in RAM and uncompress
imgs_array = [] imgs_array = []
for data in ep[f"/observations/images/{camera}"]: for data in ep[f"/observations/images/{camera}"]:
imgs_array.append(cv2.imdecode(data, 1)) imgs_array.append(cv2.cvtColor(cv2.imdecode(data, 1), cv2.COLOR_BGR2RGB))
imgs_array = np.array(imgs_array) imgs_array = np.array(imgs_array)
imgs_per_cam[camera] = imgs_array imgs_per_cam[camera] = imgs_array