yuntian-deng commited on
Commit
b887dca
·
1 Parent(s): f638c1a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -217,7 +217,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
217
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
218
  norm_x = x + (1920 - 512) / 2
219
  norm_y = y + (1080 - 512) / 2
220
- if DEBUG_TEACHER_FORCING:
221
  norm_x = x
222
  norm_y = y
223
  #action_descriptions.append(f"{(norm_x-prev_x):.0f}~{(norm_y-prev_y):.0f}")
@@ -229,7 +229,7 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
229
  x, y = pos
230
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
231
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
232
- if DEBUG_TEACHER_FORCING:
233
  norm_x = x + (1920 - 512) / 2
234
  norm_y = y + (1080 - 512) / 2
235
  #if DEBUG:
 
217
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
218
  norm_x = x + (1920 - 512) / 2
219
  norm_y = y + (1080 - 512) / 2
220
+ if False and DEBUG_TEACHER_FORCING:
221
  norm_x = x
222
  norm_y = y
223
  #action_descriptions.append(f"{(norm_x-prev_x):.0f}~{(norm_y-prev_y):.0f}")
 
229
  x, y = pos
230
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
231
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
232
+ if False and DEBUG_TEACHER_FORCING:
233
  norm_x = x + (1920 - 512) / 2
234
  norm_y = y + (1080 - 512) / 2
235
  #if DEBUG: