LiamKhoaLe commited on
Commit
74fa88b
·
1 Parent(s): 1d88641

Robot spawn on water

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -494,7 +494,7 @@ def _pipeline(uid,img_path):
494
 
495
  # 3- Robot initialization, position and navigation
496
  # find all (y,x) within movable_mask
497
- ys, xs = np.where(movable_mask)
498
  if len(ys)==0:
499
  # no travelable zone → bail out
500
  print(f"❌ [{uid}] no water to spawn on")
 
494
 
495
  # 3- Robot initialization, position and navigation
496
  # find all (y,x) within movable_mask
497
+ ys, xs = np.where(water_mask)
498
  if len(ys)==0:
499
  # no travelable zone → bail out
500
  print(f"❌ [{uid}] no water to spawn on")