update - fix build/start issues
parent
f3e83b985c
commit
2352e94e6e
|
@ -11,8 +11,8 @@ def imwrite_unicode(path, img, params=None):
|
|||
root, ext = os.path.splitext(path)
|
||||
if not ext:
|
||||
ext = ".png"
|
||||
result, encoded_img = cv2.imencode(ext, img, params if params else [])
|
||||
result, encoded_img = cv2.imencode(f".{ext}", img, params if params is not None else [])
|
||||
if result:
|
||||
encoded_img.tofile(path)
|
||||
return True
|
||||
return False
|
|
@ -10,12 +10,20 @@ psutil==5.9.8
|
|||
tk==0.1.0
|
||||
customtkinter==5.2.2
|
||||
pillow==11.1.0
|
||||
pygrabber==0.1.4
|
||||
|
||||
# Torch with CUDA for Windows/Linux and CPU-only fallback for macOS
|
||||
torch==2.5.1+cu118; sys_platform != 'darwin'
|
||||
torch==2.5.1; sys_platform == 'darwin'
|
||||
|
||||
torchvision==0.20.1; sys_platform != 'darwin'
|
||||
torchvision==0.20.1; sys_platform == 'darwin'
|
||||
|
||||
# Use only GPU-accelerated ONNX runtime
|
||||
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
|
||||
onnxruntime-gpu==1.17; sys_platform != 'darwin'
|
||||
|
||||
tensorflow; sys_platform != 'darwin'
|
||||
|
||||
opennsfw2==0.10.2
|
||||
protobuf==4.23.2
|
||||
|
|
Loading…
Reference in New Issue