yuntian-deng commited on
Commit
5f4dfaa
·
1 Parent(s): 4de630a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -101,7 +101,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
101
  #prev_x, prev_y = previous_actions[i-1][1]
102
  draw.line([prev_x, prev_y, x, y], fill=color, width=1)
103
  prev_x, prev_y = x, y
104
- draw.ellipse([x_scaled-2, y_scaled-2, x_scaled+2, y_scaled+2], fill=(0, 255, 0))
105
  #pil_image = pil_image.convert("RGB")
106
 
107
  return np.array(pil_image)
 
101
  #prev_x, prev_y = previous_actions[i-1][1]
102
  draw.line([prev_x, prev_y, x, y], fill=color, width=1)
103
  prev_x, prev_y = x, y
104
+ draw.ellipse([x_scaled*4-2, y_scaled*4-2, x_scaled*4+2, y_scaled*4+2], fill=(0, 255, 0))
105
  #pil_image = pil_image.convert("RGB")
106
 
107
  return np.array(pil_image)