summaryrefslogtreecommitdiff
path: root/recursive.py
diff options
context:
space:
mode:
Diffstat (limited to 'recursive.py')
-rw-r--r--recursive.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/recursive.py b/recursive.py
index d262ec9..73184dd 100644
--- a/recursive.py
+++ b/recursive.py
@@ -12,7 +12,8 @@ from util import html
import torch
from run_engine import run_trt_engine, run_onnx
import datetime
-from pil import Image
+from PIL import Image, ImageOps
+from shutil import copyfile, rmtree
opt = TestOptions().parse(save=False)
data_opt = DatasetOptions().parse(opt.unknown)