Как интегрировать собственные изображения в этот код машинного обучения доктора Мартина? - PullRequest
Новая
       18

Как интегрировать собственные изображения в этот код машинного обучения доктора Мартина?

0 голосов
/ 14 июня 2019

Я очарован кодом для машинного обучения доктора Мартина на Github . Этот код дает очень хорошую точность с набором данных mNist. Теперь я хотел бы использовать этот код для работы над ML с моими собственными цветными изображениями для другого приложения. Я новичок в ML, а также в Python и обучения. Я получил помощь, чтобы найти, куда данные mNist поступают в сеть.

Загрузка изображений на строка 37: # Загрузка изображений и меток в mnist.test (10K изображений + метки) и mnist.train (60K изображений + метки) mnist = mnistdata.read_data_sets ("data", one_hot = True, reshape = False)

Подача в сеть по адресу Строка 98: .... 

# You can call this function in a loop to train the model, 100 images at a time
def training_step(i, update_test_data, update_train_data):

    # training on batches of 100 images with 100 labels
    batch_X, batch_Y = mnist.train.next_batch(100)

    # compute training values for visualisation
    if update_train_data:
        a, c, im, w, b = sess.run([accuracy, cross_entropy, I, allweights, allbiases], {X: batch_X, Y_: batch_Y})
        print(str(i) + ": accuracy:" + str(a) + " loss: " + str(c) + " (lr:" + str(learning_rate) + ")")
        datavis.append_training_curves_data(i, a, c)
        datavis.update_image1(im)
        datavis.append_data_histograms(i, w, b)

Однако я до сих пор не могу понять, как это изменить, чтобы загружать и передавать свои собственные изображения в сеть.

Найден один пример на здесь .

Как я могу изменить существующую загрузку mNist с моими собственными изображениями? Я хотел бы создать свой собственный набор данных mNist с собственными изображениями, чтобы я мог использовать его с существующим кодом.

...