Merge pull request #99 from jasmcaus/master

Sanity checks + Remove unused imports
xuebinqin · Jan 24, 2021 · 527af34 · 527af34
2 parents bee18a8 + 70e6f5c
commit 527af34
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 1 deletion.
diff --git a/__pycache__/data_loader.cpython-37.pyc b/__pycache__/data_loader.cpython-37.pyc
diff --git a/model/u2net.py b/model/u2net.py
@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-from torchvision import models
 import torch.nn.functional as F
 
 class REBNCONV(nn.Module):

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+numpy==1.15.2
+scikit-image==0.14.0
+torch==0.4.1.post2
+torchvision==0.2.1
+pillow==7.2.0
diff --git a/u2net_train.py b/u2net_train.py
@@ -11,6 +11,7 @@
 
 import numpy as np
 import glob
+import os
 
 from data_loader import Rescale
 from data_loader import RescaleT