Spaces:
Running
Running
gavinyuan
commited on
Commit
·
afe1649
1
Parent(s):
1d384c2
update: GPEN weights
Browse files
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.
|
275 |
elif landmark_smooth == 'savgol':
|
276 |
-
target_lmks = savgol_filter_landmark(target_lmks).astype(np.
|
277 |
elif landmark_smooth == 'cancel':
|
278 |
-
target_lmks = target_lmks.astype(np.
|
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 '''
|