How to Save files as Original file name using python?

Asked Jul 10 '21 at 03:42

Active Jul 10 '21 at 03:42

Viewed 571 times

Hi I want to save the predicted image in a folder using the same name as the original but my cv2.imwrite is not saving the images in the folder.

path2=r'/data/folder/'
def get_segmented(image):
    for i, file in enumerate(glob.glob(image)):
        
        print(file,i)
        start_time = time()
        image = image_loader(file)
    
        output = model(image)
        output = output.detach().cpu()
        output = np.resize(output, (160, 160))
        
        img_filename = os.path.splitext(file)[0]
        print(cv2.imwrite(path2 + img_filename + '.png',output*255))

asked Jul 10 '21 at 03:42

Hammad Younas

Why is there a print at the end? Have you tried avoiding print? I would try with just `cv2.imwrite(path2 + img_filename + '.png',output*255)`. – DeusDev Jul 10 '21 at 03:49
Just to see the output is saving or not – Hammad Younas Jul 10 '21 at 03:56
What is your problem, saving file with original name or saving file? – imxitiz Jul 10 '21 at 03:57
saving with the original name. when I run this code cv2.imwrite output is False – Hammad Younas Jul 10 '21 at 04:00
What you you providing in `image` argument? and why are you implementing loop for there? and also are you sure that your `output` is of `numpy.ndarray` class? – imxitiz Jul 10 '21 at 04:12
image=path of images – Hammad Younas Jul 10 '21 at 04:15
Try to visualize final output image using `matplotlib` https://matplotlib.org/stable/api/_as_gen/matplotlib.pyplot.imshow.html https://stackoverflow.com/a/35287898/12446721 – imxitiz Jul 10 '21 at 04:19
Ok, thanks, I have solved the issue, the output is numpy.ndarray class – Hammad Younas Jul 10 '21 at 04:31

How to Save files as Original file name using python?

0 Answers0