簡體   English   中英

如何使用 Tensorflow 從安裝的谷歌驅動器加載谷歌 Colab 筆記本中的圖像

[英]How to load images in Google Colab notebook using Tensorflow from mounted Google drive

在 Google Colab 筆記本中,我安裝了 Google 驅動器並且可以查看我的文件。 我正在嘗試加載一個壓縮目錄,該目錄有兩個文件夾,每個文件夾中有幾個圖片文件。

我遵循了 Tensorflow 站點上的一個示例,該示例有一個關於如何加載圖片的示例,但它使用的是遠程位置。 這是網站 - https://www.tensorflow.org/tutorials/load_data/images

這是示例中有效的代碼:

data_root_orig = tf.keras.utils.get_file(origin='https://storage.googleapis.com/download.tensorflow.org/example_images/flower_photos.tgz',
                                        fname='flower_photos', untar=True)
data_root = pathlib.Path(data_root_orig)
print(data_root)

這是我嘗試從已安裝的 Google 驅動器引用壓縮目錄的修訂代碼:

data_root_orig = tf.keras.utils.get_file(origin='/content/gdrive/My Drive/TrainingPictures/',
                                         fname='TrainingPictures_Car', untar=True)
data_root = pathlib.Path(data_root_orig)
print(data_root)

我收到此錯誤:

ValueError: unknown url type: '/content/gdrive/My Drive/TrainingPictures/'

它顯然需要一個 URL 而不是我提供的路徑。 我想知道如何加載 Google 驅動器提供的壓縮目錄。

在這種情況下,不需要使用tf.keras.utils.get_file() ,只有Path就足夠了。

這里有 2 種方法來做到這一點

首先: !unzip -q 'content/gdrive/My Drive/TrainingPictures/TrainingPictures_Car.zip'

它將被解壓到“/content/”

import pathlib
data = pathlib.Path('/content/folders_inside_zip') 
count = len(list(data.glob('*/*.jpg')))
count

第二:如果存檔已經在谷歌驅動器中解壓縮:

import pathlib
data = pathlib.Path('/content/gdrive/My Drive/TrainingPictures/') 
count = len(list(data.glob('*.jpg')))
count

在我的情況下,它實際上是通過刪除所有導入和庫並將路徑設置為字符串來工作的。 該文件必須上傳到 google colab。

content_path = "cat.jpg"

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM