LiamKhoaLe commited on
Commit
4b3eff9
·
1 Parent(s): 8e7b586

Rm mask [255, 9, 92]

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -55,7 +55,7 @@ ade_palette = np.array([
55
  [4, 250, 7], [224, 5, 255], [235, 255, 7], [150, 5, 61], [120, 120, 70],
56
  [8, 255, 51], [255, 6, 82], [143, 255, 140], [204, 255, 4], [255, 51, 7],
57
  [204, 70, 3], [0, 102, 200], [61, 230, 250], [255, 6, 51], [11, 102, 255],
58
- [255, 7, 71], [255, 9, 224], [9, 7, 230], [220, 220, 220], [255, 9, 92],
59
  [112, 9, 255], [8, 255, 214], [7, 255, 224], [255, 184, 6], [10, 255, 71],
60
  [255, 41, 10], [7, 255, 255], [224, 255, 8], [102, 8, 255], [255, 61, 6],
61
  [255, 194, 7], [255, 122, 8], [0, 255, 20], [255, 8, 41], [255, 5, 153],
@@ -509,7 +509,7 @@ def _pipeline(uid,img_path):
509
  robot = Robot(SPRITE)
510
  # Robot will be spawn on the closest movable mask to top-left
511
  robot.pos = [spawn_x, spawn_y]
512
- path = knn_path(robot.pos, centres, water_mask)
513
 
514
  # 4- Video synthesis
515
  out_tmp=f"{OUTPUT_DIR}/{uid}_tmp.mp4"
 
55
  [4, 250, 7], [224, 5, 255], [235, 255, 7], [150, 5, 61], [120, 120, 70],
56
  [8, 255, 51], [255, 6, 82], [143, 255, 140], [204, 255, 4], [255, 51, 7],
57
  [204, 70, 3], [0, 102, 200], [61, 230, 250], [255, 6, 51], [11, 102, 255],
58
+ [255, 7, 71], [255, 9, 224], [9, 7, 230], [220, 220, 220],
59
  [112, 9, 255], [8, 255, 214], [7, 255, 224], [255, 184, 6], [10, 255, 71],
60
  [255, 41, 10], [7, 255, 255], [224, 255, 8], [102, 8, 255], [255, 61, 6],
61
  [255, 194, 7], [255, 122, 8], [0, 255, 20], [255, 8, 41], [255, 5, 153],
 
509
  robot = Robot(SPRITE)
510
  # Robot will be spawn on the closest movable mask to top-left
511
  robot.pos = [spawn_x, spawn_y]
512
+ path = knn_path(robot.pos, centres, movable_mask)
513
 
514
  # 4- Video synthesis
515
  out_tmp=f"{OUTPUT_DIR}/{uid}_tmp.mp4"