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

fix project layout

Browse Source
This commit is contained in:
Daniel Gatis 2022-01-09 15:54:30 -03:00
parent eba6b2dbf3
commit 5fe7858a55
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rembg/detect.py
View File

@ -16,7 +16,7 @@ from torchvision import transforms
from tqdm import tqdm from tqdm import tqdm
from .data_loader import RescaleT, ToTensorLab from .data_loader import RescaleT, ToTensorLab
from .u2net import U2NETP, U2NET from .u2net import U2NET, U2NETP
def download_file_from_google_drive(id, fname, destination): def download_file_from_google_drive(id, fname, destination):

6
setup.py
View File

@ -1,12 +1,12 @@
import os
import pathlib import pathlib
import sys import sys
import os
sys.path.append(os.path.dirname(__file__)) sys.path.append(os.path.dirname(__file__))
import versioneer
from setuptools import find_packages, setup from setuptools import find_packages, setup
import versioneer
here = pathlib.Path(__file__).parent.resolve() here = pathlib.Path(__file__).parent.resolve()
long_description = (here / "README.md").read_text(encoding="utf-8") long_description = (here / "README.md").read_text(encoding="utf-8")