Make a detection directory if there isn't one with the name

Author: ayooshkathuria

det_messi.jpg

@@ -113,6 +113,9 @@ def arg_parse():
         print ("No file or directory with the name {}".format(images))
         exit()
 
+    if not os.path.exists(args.det):
+        os.makedirs(args.det)
+        
     load_batch = time.time()
 
     batches = list(map(prep_image, imlist, [inp_dim for x in range(len(imlist))]))
@@ -122,6 +125,7 @@ def arg_parse():
     im_dim_list = torch.FloatTensor(im_dim_list).repeat(1,2)
 
 
+    
     if CUDA:
         im_dim_list = im_dim_list.cuda()