yuntian-deng commited on
Commit
e712a69
·
1 Parent(s): 08c04fc

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -4
main.py CHANGED
@@ -216,8 +216,10 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
216
  x, y = pos
217
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
218
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
219
- norm_x = x + (1920 - 512) / 2
220
- norm_y = y + (1080 - 512) / 2
 
 
221
  if False and DEBUG_TEACHER_FORCING:
222
  norm_x = x
223
  norm_y = y
@@ -230,8 +232,10 @@ def predict_next_frame(previous_frames: List[np.ndarray], previous_actions: List
230
  x, y = pos
231
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
232
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
233
- norm_x = x + (1920 - 512) / 2
234
- norm_y = y + (1080 - 512) / 2
 
 
235
  if False and DEBUG_TEACHER_FORCING:
236
  norm_x = x #+ (1920 - 512) / 2
237
  norm_y = y #+ (1080 - 512) / 2
 
216
  x, y = pos
217
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
218
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
219
+ #norm_x = x + (1920 - 512) / 2
220
+ #norm_y = y + (1080 - 512) / 2
221
+ norm_x = x
222
+ norm_y = y
223
  if False and DEBUG_TEACHER_FORCING:
224
  norm_x = x
225
  norm_y = y
 
232
  x, y = pos
233
  #norm_x = int(round(x / 256 * 1024)) #x + (1920 - 256) / 2
234
  #norm_y = int(round(y / 256 * 640)) #y + (1080 - 256) / 2
235
+ #norm_x = x + (1920 - 512) / 2
236
+ #norm_y = y + (1080 - 512) / 2
237
+ norm_x = x
238
+ norm_y = y
239
  if False and DEBUG_TEACHER_FORCING:
240
  norm_x = x #+ (1920 - 512) / 2
241
  norm_y = y #+ (1080 - 512) / 2