diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 11:37:18 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-05-10 11:37:18 +0300 |
| commit | a5121e7a0623db328a9462d340d389ed6737374a (patch) | |
| tree | 2fa6051d457555ef2d61793af2756a44c0ea221c /modules/codeformer_model.py | |
| parent | 550256db1ce18778a9d56ff343d844c61b9f9b83 (diff) | |
fixes for B007
Diffstat (limited to 'modules/codeformer_model.py')
| -rw-r--r-- | modules/codeformer_model.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/modules/codeformer_model.py b/modules/codeformer_model.py index 8e56cb89..ececdbae 100644 --- a/modules/codeformer_model.py +++ b/modules/codeformer_model.py @@ -94,7 +94,7 @@ def setup_model(dirname): self.face_helper.get_face_landmarks_5(only_center_face=False, resize=640, eye_dist_threshold=5)
self.face_helper.align_warp_face()
- for idx, cropped_face in enumerate(self.face_helper.cropped_faces):
+ for cropped_face in self.face_helper.cropped_faces:
cropped_face_t = img2tensor(cropped_face / 255., bgr2rgb=True, float32=True)
normalize(cropped_face_t, (0.5, 0.5, 0.5), (0.5, 0.5, 0.5), inplace=True)
cropped_face_t = cropped_face_t.unsqueeze(0).to(devices.device_codeformer)
|