Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make CLIP interrogate ranks output sane values

This commit is contained in:
AUTOMATIC 2022-10-17 08:41:02 +03:00
parent 5c94aaf290
commit b99d3cf6dd
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/interrogate.py
View File

@ -123,7 +123,7 @@ class InterrogateModels:
return caption[0] return caption[0]
def interrogate(self, pil_image, include_ranks=False): def interrogate(self, pil_image):
res = None res = None
try: try:
@ -156,8 +156,8 @@ class InterrogateModels:
for name, topn, items in self.categories: for name, topn, items in self.categories:
matches = self.rank(image_features, items, top_count=topn) matches = self.rank(image_features, items, top_count=topn)
for match, score in matches: for match, score in matches:
if include_ranks: if shared.opts.interrogate_return_ranks:
res += f", ({match}:{score})" res += f", ({match}:{score/100:.3f})"
else: else:
res += ", " + match res += ", " + match