test

1 files changed, 1 insertions, 1 deletions

diff --git a/test.py b/test.py
index 2a60a65..e14d228 100644
--- a/test.py
+++ b/test.py
@@ -17,7 +17,7 @@ if __name__ == '__main__':
     opt.batchSize = 1  # test code only supports batchSize = 1
     opt.serial_batches = True  # no shuffle
     opt.no_flip = True  # no flip
-    opt.experiment = opt.start_img.split(".")[0]
+    opt.experiment = opt.start_img.split("/")[-1].split(".")[0]
 
     render_dir = opt.results_dir + opt.name + "/exp:" + opt.experiment + "/"