aboutsummaryrefslogtreecommitdiff
path: root/modules/textual_inversion/autocrop.py
diff options
context:
space:
mode:
authorAUTOMATIC <16777216c@gmail.com>2023-06-09 22:48:18 +0300
committerAUTOMATIC <16777216c@gmail.com>2023-06-09 22:48:18 +0300
commit3b11f17a374520e493e120e7f47443acd97393c8 (patch)
tree743d96990f646fb49ff377a67849c288c31e177e /modules/textual_inversion/autocrop.py
parentbaf6946e06249c5af9851c60171692c44ef633e0 (diff)
parent59419bd64a1581caccaac04dceb66c1c069a2db1 (diff)
Merge branch 'dev' into release_candidate
Diffstat (limited to 'modules/textual_inversion/autocrop.py')
-rw-r--r--modules/textual_inversion/autocrop.py14
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 []