wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge 8b9b2d2e08 into b568de35ad

This commit is contained in:
Jaco Cronje 2025-03-28 12:44:37 -05:00 committed by GitHub
parent b568de35ad 8b9b2d2e08
commit b8ceda18c6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lerobot/scripts/control_robot.py
View File

@ -134,6 +134,13 @@ python lerobot/scripts/control_robot.py \
``` ```
""" """
import cv2
# Make sure that the UI gets initialized before PyAV (av) gets imported by torchvision
# This solves the hanging issue with cv2.imshow on Ubuntu
cv2.namedWindow("i")
cv2.destroyAllWindows()
import logging import logging
import time import time
from dataclasses import asdict from dataclasses import asdict

6
lerobot/scripts/control_sim_robot.py
View File

@ -88,6 +88,12 @@ import time
from pathlib import Path from pathlib import Path
import cv2 import cv2
# Make sure that the UI gets initialized before PyAV (av) gets imported by torchvision
# This solves the hanging issue with cv2.imshow on Ubuntu
cv2.namedWindow("i")
cv2.destroyAllWindows()
import gymnasium as gym import gymnasium as gym
import numpy as np import numpy as np
import torch import torch