diff options
author | AUTOMATIC <16777216c@gmail.com> | 2023-06-09 19:48:18 +0000 |
---|---|---|
committer | AUTOMATIC <16777216c@gmail.com> | 2023-06-09 19:48:18 +0000 |
commit | 3b11f17a374520e493e120e7f47443acd97393c8 (patch) | |
tree | 743d96990f646fb49ff377a67849c288c31e177e /modules/textual_inversion/autocrop.py | |
parent | baf6946e06249c5af9851c60171692c44ef633e0 (diff) | |
parent | 59419bd64a1581caccaac04dceb66c1c069a2db1 (diff) | |
download | stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.tar.gz stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.tar.bz2 stable-diffusion-webui-gfx803-3b11f17a374520e493e120e7f47443acd97393c8.zip |
Merge branch 'dev' into release_candidate
Diffstat (limited to 'modules/textual_inversion/autocrop.py')
-rw-r--r-- | modules/textual_inversion/autocrop.py | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/modules/textual_inversion/autocrop.py b/modules/textual_inversion/autocrop.py index 8e667a4d..75705459 100644 --- a/modules/textual_inversion/autocrop.py +++ b/modules/textual_inversion/autocrop.py @@ -77,27 +77,27 @@ def focal_point(im, settings): pois = []
weight_pref_total = 0
- if len(corner_points) > 0:
+ if corner_points:
weight_pref_total += settings.corner_points_weight
- if len(entropy_points) > 0:
+ if entropy_points:
weight_pref_total += settings.entropy_points_weight
- if len(face_points) > 0:
+ if face_points:
weight_pref_total += settings.face_points_weight
corner_centroid = None
- if len(corner_points) > 0:
+ if corner_points:
corner_centroid = centroid(corner_points)
corner_centroid.weight = settings.corner_points_weight / weight_pref_total
pois.append(corner_centroid)
entropy_centroid = None
- if len(entropy_points) > 0:
+ if entropy_points:
entropy_centroid = centroid(entropy_points)
entropy_centroid.weight = settings.entropy_points_weight / weight_pref_total
pois.append(entropy_centroid)
face_centroid = None
- if len(face_points) > 0:
+ if face_points:
face_centroid = centroid(face_points)
face_centroid.weight = settings.face_points_weight / weight_pref_total
pois.append(face_centroid)
@@ -187,7 +187,7 @@ def image_face_points(im, settings): except Exception:
continue
- if len(faces) > 0:
+ if faces:
rects = [[f[0], f[1], f[0] + f[2], f[1] + f[3]] for f in faces]
return [PointOfInterest((r[0] +r[2]) // 2, (r[1] + r[3]) // 2, size=abs(r[0]-r[2]), weight=1/len(rects)) for r in rects]
return []
|