x/themblem
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

common.py: fix predict

Browse Source
This commit is contained in:
Fam Zheng 2025-12-27 12:38:02 +00:00
parent ad4ca27b7a
commit a04471d574
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
emblem5/ai/common.py
View File

@ -517,7 +517,7 @@ def verify_frame(model, transforms, frame_img, orig_img):
side_by_side_img = side_by_side_img.convert('RGB') side_by_side_img = side_by_side_img.convert('RGB')
with tempfile.NamedTemporaryFile(suffix='.jpg') as f: with tempfile.NamedTemporaryFile(suffix='.jpg') as f:
side_by_side_img.save(f.name) side_by_side_img.save(f.name)
return predict(model, transforms, Image.open(f.name).convert('RGB')) return predict(model, transforms, Image.open(f.name).convert('RGB'), ncells=3)
def parse_ranges(s): def parse_ranges(s):
ret = [] ret = []