Spaces:

xinjjj
/

ImgRoboAssetGen

Running on Zero

xinjie.wang commited on Apr 14

Commit

eca50d9

1 Parent(s): 2a08301

update

Files changed (1) hide show

common.py CHANGED Viewed

@@ -567,6 +567,7 @@ def extract_urdf(
     for checker in CHECKERS:
         images = image_paths
         if isinstance(checker, ImageSegChecker):
             images = [IMAGE_BUFFER["raw_image"], IMAGE_BUFFER["seg_image"]]
         images_list.append(images)

     for checker in CHECKERS:
         images = image_paths
         if isinstance(checker, ImageSegChecker):
+            print("IMAGE_BUFFER", IMAGE_BUFFER.keys())
             images = [IMAGE_BUFFER["raw_image"], IMAGE_BUFFER["seg_image"]]
         images_list.append(images)