gavinyuan commited on
Commit
afe1649
1 Parent(s): 1d384c2

update: GPEN weights

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -271,11 +271,11 @@ def process_video(
271
  if landmark_smooth == 'kalman':
272
  target_lmks = kalman_filter_landmark(target_lmks,
273
  process_noise=0.01,
274
- measure_noise=0.01).astype(np.int)
275
  elif landmark_smooth == 'savgol':
276
- target_lmks = savgol_filter_landmark(target_lmks).astype(np.int)
277
  elif landmark_smooth == 'cancel':
278
- target_lmks = target_lmks.astype(np.int)
279
  else:
280
  raise KeyError('Not supported landmark_smooth choice')
281
  ''' Crop source image '''
 
271
  if landmark_smooth == 'kalman':
272
  target_lmks = kalman_filter_landmark(target_lmks,
273
  process_noise=0.01,
274
+ measure_noise=0.01).astype(np.int32)
275
  elif landmark_smooth == 'savgol':
276
+ target_lmks = savgol_filter_landmark(target_lmks).astype(np.int32)
277
  elif landmark_smooth == 'cancel':
278
+ target_lmks = target_lmks.astype(np.int32)
279
  else:
280
  raise KeyError('Not supported landmark_smooth choice')
281
  ''' Crop source image '''