From 6b9d4a6f6c36635088ccea6add22c14c0ae6a96f Mon Sep 17 00:00:00 2001 From: WiseDoge Date: Sun, 13 Jan 2019 16:30:52 +0800 Subject: [PATCH] Fix a bug on Windows when saving an output img. --- cam_demo.py | 2 +- detect.py | 9 +++++++-- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/cam_demo.py b/cam_demo.py index b128be93..23355978 100644 --- a/cam_demo.py +++ b/cam_demo.py @@ -116,7 +116,7 @@ def arg_parse(): img, orig_im, dim = prep_image(frame, inp_dim) -# im_dim = torch.FloatTensor(dim).repeat(1,2) + im_dim = torch.FloatTensor(dim).repeat(1,2) if CUDA: diff --git a/detect.py b/detect.py index 4d6c5e76..89e1e385 100644 --- a/detect.py +++ b/detect.py @@ -15,6 +15,8 @@ import random import pickle as pkl import itertools +import platform + class test_net(nn.Module): def __init__(self, num_layers, input_size): @@ -298,8 +300,11 @@ def write(x, batches, results): list(map(lambda x: write(x, im_batches, orig_ims), output)) - - det_names = pd.Series(imlist).apply(lambda x: "{}/det_{}".format(args.det,x.split("/")[-1])) + + if platform.system() == 'Windows': + det_names = pd.Series(imlist).apply(lambda x: "{}\\det_{}".format(args.det,x.split("\\")[-1])) + else: + det_names = pd.Series(imlist).apply(lambda x: "{}/det_{}".format(args.det,x.split("/")[-1])) list(map(cv2.imwrite, det_names, orig_ims))