From 1af3f87c6143dd82fa650c36836ea87fe6114071 Mon Sep 17 00:00:00 2001 From: julenalbi Date: Fri, 4 Nov 2022 10:35:08 +0100 Subject: [PATCH] make inference script platform agnostic --- inference.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/inference.py b/inference.py index af835be..30f216f 100644 --- a/inference.py +++ b/inference.py @@ -3,7 +3,6 @@ import argparse import os import sys -import time import scipy.io as sio from PIL import Image @@ -58,7 +57,7 @@ def load(saver, sess, ckpt_path): def main(): args = get_arguments() - filename = args.img_path.split('/')[-1] + filename = os.path.basename(args.img_path) file_type = filename.split('.')[-1] if os.path.isfile(args.img_path): @@ -119,9 +118,9 @@ def main(): im = Image.fromarray(msk[0]) if not os.path.exists(SAVE_DIR): os.makedirs(SAVE_DIR) - im.save(SAVE_DIR + filename) + im.save(os.path.join(SAVE_DIR + filename)) - print('The output file has been saved to {0}'.format(SAVE_DIR + filename)) + print('The output file has been saved to {0}'.format(os.path.join(SAVE_DIR + filename))) if __name__ == '__main__':