Загрузить изображения с метками в Google Collab - PullRequest
Новая
       49

Загрузить изображения с метками в Google Collab

0 голосов
/ 21 мая 2018

Я использую ноутбук Jupyter в Google Collab.Мой обучающий набор данных выглядит так: 

/data/label1/img1.jpeg
.
.
.
/data/label2/img90.jpeg

Я хочу импортировать такой набор данных.Вещи, которые я пробовал

Шаг 1: 

!pip install -U -q PyDrive
%matplotlib inline
import matplotlib
import matplotlib.pyplot as plt
from os import walk
import os
from pydrive.auth import GoogleAuth
from pydrive.drive import GoogleDrive
from google.colab import auth
from oauth2client.client import GoogleCredentials

Шаг 2: 

# 1. Authenticate and create the PyDrive client.
auth.authenticate_user()
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default()
drive = GoogleDrive(gauth)

Шаг 3 

file_to_download = os.path.expanduser('./data/')
file_list = drive.ListFile(
    {'q': 'id_of_the_data_directory'})

Не уверен, что делать дальше,Папка data является моей записной папкой на диске.Я хочу прочитать изображения вместе с метками. Чтобы сделать то же самое, я использую код: 

filename_queue=tf.train.string_input_producer(tf.train.match_filenames_once('data/*/*.jpeg'))
image_reader=tf.WholeFileReader()
key,image_file=image_reader.read(filename_queue)
#key is the entire path to the jpeg file and we need only the subfolder as the label
S = tf.string_split([key],'\/')
length = tf.cast(S.dense_shape[1],tf.int32)
label = S.values[length-tf.constant(2,dtype=tf.int32)]
label = tf.string_to_number(label,out_type=tf.int32)
#decode the image
image=tf.image.decode_jpeg(image_file)
#then code to place labels and folders in corresponding arrays

Ответы [ 2 ]

0 голосов
/ 25 мая 2018

Вы должны загрузить свой набор данных рекурсивным способом. Здесь - пример загрузки набора данных с вашего Google Диска в Colab

.
0 голосов
/ 21 мая 2018

Прежде всего я хочу отметить, что мы не можем получить доступ к папке напрямую.Нам нужно установить точку монтирования и через нее получить доступ ко всему содержимому диска.Благодаря этому answer Следуйте инструкциям, приведенным выше.Но просто убедитесь, что вы изменили свой путь в соответствии с созданной новой папкой диска.

PS: Я все еще оставил вопрос открытым, потому что вы можете добраться до него с помощью набора данных изображений с именами подпапок в качестве меток обучающих изображений, этоработает, поэтому решение, опубликованное здесь, работает как для каталогов с подпапками, так и для каталогов с файлами.

...