Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

jumbo merge

Browse Source
This commit is contained in:
Vladimir Mandic
2023-06-13 11:58:14 -04:00
parent 0d101b9def
commit cb307399dd
72 changed files with 535 additions and 408 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/codeformer_model.py
+1 -1
View File
@@ -91,7 +91,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)