1. 程式人生 > >python-判斷檔案/圖片路徑是否存在

python-判斷檔案/圖片路徑是否存在

1.在向資料夾中儲存資料前,先判斷該資料夾(路徑)是否存在

save_path = '/root/.../image/result'
if not os.path.exists(save_path):
    os.makedirs(save_path)

本來路徑裡只有到image資料夾的,執行完後會自動在image下建立result資料夾

2.在開啟某些圖片時,可以先判斷該圖片的路徑是否存在

if img == 'target7.jpg':
    img_path = os.path.join(dir_path, img)
    if imghdr.what(img_path):  # 判斷是否能開啟
        target_img = Image.open(img_path)
        target_img.save(os.path.join(root, 'trainB/{}_B.jpg'.format(count+6)))