mod utils

2 files changed, 19 insertions, 5 deletions

diff --git a/check/app/utils/file_utils.py b/check/app/utils/file_utils.py
index 5c7b39d..1ed1833 100644
--- a/check/app/utils/file_utils.py
+++ b/check/app/utils/file_utils.py
@@ -310,6 +310,25 @@ def mkdirs(fp):
   fpp = fpp.parent if fpp.suffix else fpp
   fpp.mkdir(parents=True, exist_ok=True)
 
+def ensure_posixpath(fp):
+  """Ensures filepath is pathlib.Path
+  :param fp: a (str, LazyFile, PosixPath)
+  :returns: a PosixPath filepath object
+  """
+  if type(fp) == str:
+    fpp = Path(fp)
+  elif type(fp) == click.utils.LazyFile:
+    fpp = Path(fp.name)
+  elif type(fp) == pathlib.PosixPath:
+    fpp = fp
+  else:
+    raise TypeError('{} is not a valid filepath type'.format(type(fp)))
+  return fpp
+
+def ensure_dir(fp):
+  if not Path(fp).is_dir():
+    mkdirs(fp)
+
 
 def ext_media_format(ext):
   """Converts file extension into Enum MediaType
diff --git a/check/app/utils/im_utils.py b/check/app/utils/im_utils.py
index 747e900..1d1affb 100644
--- a/check/app/utils/im_utils.py
+++ b/check/app/utils/im_utils.py
@@ -11,12 +11,7 @@ from skimage import feature
 import imutils
 import time
 import numpy as np
-import torch
-import torch.nn as nn
-import torchvision.models as models
-import torchvision.transforms as transforms
 import struct
-from torch.autograd import Variable
 from sklearn.metrics.pairwise import cosine_similarity
 import datetime