From 46f61c05bd1651686ed747a0c9a066842fe638fb Mon Sep 17 00:00:00 2001
From: Johnson yue <Johnson-yue@users.noreply.github.com>
Date: Mon, 30 Nov 2020 12:17:46 +0800
Subject: [PATCH] add os module import

---
 u2net_train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/u2net_train.py b/u2net_train.py
index 40395e94..6f62aa31 100644
--- a/u2net_train.py
+++ b/u2net_train.py
@@ -10,7 +10,7 @@
 import torchvision.transforms as standard_transforms
 
 import numpy as np
-import glob
+import glob,os 
 
 from data_loader import Rescale
 from data_loader import RescaleT