如何將 Kaggle 中輸出的圖片結果下載到本地？

使用 GPU 加速深度學習的訓練是很關鍵的，對於缺少計算資源的人來說，在 Kaggle 上使用 GPU 訓練模型是一個相對不錯的體驗。但是，如果，你的輸出結果是圖片，那該如下將訓練後的圖片下載到本地呢?

我嘗試了很多辦法，最終想到一個相對不錯的點子：將輸出的圖片打包為 ZIP 文件。具體操作如下：

上傳需要訓練的模型與數據(以 ZIP 的形式)：https://www.kaggle.com/xinzone/demotest

然後以此數據集創建一個 kernel：

選中 Notebook：

這樣便可以設置使用 GPU：

使用調試你的代碼，切換到數據集與項目所在目錄：

cd ../input/gcanet/

運行代碼

!python test.py --task dehaze --gpu_id 0 --indir examples --outdir /kaggle/working/imgs

這裡需要注意的是你的項目和數據是只讀的，無法向其中寫入其他內容，因而，需要將圖片的輸出放在可以進行寫操作的 kaggle/working 目錄中，即 參數 --outdir /kaggle/working/imgs。將輸出的圖片放在 imgs 下面。

這樣，如果直接 Commit 在你的 kaggle 頁面將可以看到 imgs 下的圖片的可視化，但是無法下載。

為了可以下載，你需要將 imgs 打包為 ZIP 文件：

from pathlib import Path

import zipfile

img_root = Path("/kaggle/working/imgs")

with zipfile.ZipFile("imgs.zip", "w") as z:

for img_name in img_root.iterdir():

z.write(img_name)

這樣，在 /kaggle/working 下面便多了 imgs.zip 文件：

ls /kaggle/working

輸出：

__notebook_source__.ipynb imgs/ imgs.zip

然後，將最終結果進行提交：

這樣，便可以下載輸出的圖片了：

最終的成果展示，可以直接查看我的 Kaggle，如果覺得有用，可以 Fork 並點贊。

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！