pytorch-yolo-v3
pytorch-yolo-v3 copied to clipboard
Picture saving bug fixed.
Picture output fails in Windows system, since the path is not correctly obtained.
det_names = pd.Series(imlist).apply(lambda x: "{}/det_{}".format(args.det, x.split("/")[-1]))
This trouble result from that Windows use '' as seperator, which can be easily solved by adding another split.
det_names = pd.Series(imlist).apply(lambda x: "{}/det_{}".format(args.det, (x.split("\\")[-1]).split('/')[-1]))
Hi, thanks a lot for your code @WannaSeaU, but I still have problems processing images in a directory in Windows system. When I run the code using "python detect.py --images imgs --det det" in the Anaconda prompt, where "imgs" is the directory storing many images, the detection results can only be shown in the prompt line by line without any result images being saved into "det". Can you please help me? Thank you very much.