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 '''
|