From cc5d70f6e0047eeb42828d3b1dfcb464a47365c4 Mon Sep 17 00:00:00 2001
From: Jason Dsouza <61184127+jasmcaus@users.noreply.github.com>
Date: Tue, 24 Nov 2020 16:03:41 +0530
Subject: [PATCH] Import undefined library (os)

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

diff --git a/u2net_train.py b/u2net_train.py
index 40395e9..24466d4 100644
--- a/u2net_train.py
+++ b/u2net_train.py
@@ -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