AI/rembg
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/danielgatis/rembg synced 2025-08-13 23:16:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix linters

Browse Source
This commit is contained in:
Daniel Gatis 2024-03-12 19:47:49 -03:00
parent d63da2e6fb
commit 00e3e257fe
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
rembg/bg.py
View File

@ -3,6 +3,7 @@ from enum import Enum
from typing import Any, List, Optional, Tuple, Union from typing import Any, List, Optional, Tuple, Union
import numpy as np import numpy as np
import onnxruntime as ort
from cv2 import ( from cv2 import (
BORDER_DEFAULT, BORDER_DEFAULT,
MORPH_ELLIPSE, MORPH_ELLIPSE,
@ -22,8 +23,6 @@ from .session_factory import new_session
from .sessions import sessions_class from .sessions import sessions_class
from .sessions.base import BaseSession from .sessions.base import BaseSession
import onnxruntime as ort
ort.set_default_logger_severity(3) ort.set_default_logger_severity(3)
kernel = getStructuringElement(MORPH_ELLIPSE, (3, 3)) kernel = getStructuringElement(MORPH_ELLIPSE, (3, 3))