AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-17 09:05:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix docker

Browse Source
This commit is contained in:
Daniel Gatis 2023-06-29 21:15:20 -03:00
parent b527af6af5
commit a9026d02e0
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
rembg/sessions/u2net_custom.py
View File

@ -2,6 +2,7 @@ import os
from typing import List from typing import List
import numpy as np import numpy as np
import onnxruntime as ort
import pooch import pooch
from PIL import Image from PIL import Image
from PIL.Image import Image as PILImage from PIL.Image import Image as PILImage