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

Merge 8b9b2d2e08 into 1c873df5c0

This commit is contained in:
Jaco Cronje 2025-04-04 12:16:38 +02:00 committed by GitHub
parent 1c873df5c0 8b9b2d2e08
commit 64f690f9c3
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 time
from dataclasses import asdict

6
lerobot/scripts/control_sim_robot.py
View File

@ -88,6 +88,12 @@ import time
from pathlib import Path
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 numpy as np
import torch