diff --git a/utils.py b/utils.py
index d687ef2..f990b53 100644
--- a/utils.py
+++ b/utils.py
@@ -1,6 +1,7 @@
 import cv2
 import torch
 import numpy as np
+import os
 from collections import OrderedDict
 import torchvision.transforms as transforms
 from PIL import Image