AI/U-2-Net
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/xuebinqin/U-2-Net synced 2025-08-05 18:30:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Import undefined library (os)

Browse Source
This commit is contained in:
Jason Dsouza 2020-11-24 16:03:41 +05:30 committed by GitHub
parent 379156dc46
commit cc5d70f6e0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
u2net_train.py
View File

@ -11,6 +11,7 @@ import torchvision.transforms as standard_transforms
import numpy as np
import glob
import os
from data_loader import Rescale
from data_loader import RescaleT