Кафе всегда возвращает один ярлык - PullRequest
Новая
       47

Кафе всегда возвращает один ярлык

0 голосов
/ 01 октября 2018

Я обучил модель с помощью инструментов caffe в bin, и теперь я пытаюсь провести тестирование с использованием скрипта python, я читаю изображение и сам его препроцессирую (как я делал для своего набора данных обучения), и я загружаю предварительно подготовленные веса внет, но я почти всегда (99,99% времени) получаю один и тот же результат -0- для каждого тестового изображения.Я считал, что моя модель может быть перегружена, но после обучения нескольких моделей я понял, что ярлыки, которые я получаю из прогнозов, скорее всего, являются причиной.Я также увеличил отсев и выбрал случайные посевы, чтобы преодолеть перенапряжение, и у меня есть около 60 тыс. Для обучения.Набор данных также примерно сбалансирован.Я получаю точность от 77 до 87 на этапе оценки (в зависимости от того, как я обрабатываю данные, какую архитектуру я использую и т. Д.)

Извините, мой супер-хакерский код, я некоторое время не проводил тестирование, поэтому яподозреваю, что проблема в том, как я передаю входные данные в сеть, но я не могу это понять: 

import h5py, os
import sys
sys.path.append("/home/X/Desktop/caffe-caffe-0.16/python")
from caffe.io import oversample
from caffe.io import resize_image
import caffe
from random import randint
import numpy as np
import cv2
import matplotlib.pyplot as plt
from collections import Counter as Cnt

meanImg = cv2.imread('/home/caffe/data/Ch/Final_meanImg.png')

model_def = '/home/X/Desktop/caffe-caffe-0.16/models/bvlc_googlenet/deploy.prototxt'
model_weights = '/media/X/DATA/SDet/Google__iter_140000.caffemodel'

# load the model
#caffe.set_mode_gpu()
#caffe.set_device(0)

net = caffe.Net(model_def,      # defines the structure of the model
                model_weights,  # contains the trained weights
                caffe.TEST)     # use test mode (e.g., don't perform dropout)


with open( '/home/caffe/examples/sdet/SDet/test_random.txt', 'r' ) as T, open('/media/X/DATA/SDet/results/testResults.txt','w') as testResultsFile:


    readImgCounter = 0
    runningCorrect = 0
    runningAcc = 0.0

    #testResultsFile.write('filename'+' '+'prediction'+' '+'GT')
    lines = T.readlines()

    for i,l in enumerate(lines):


        sp = l.split(' ')

        video = sp[0].split('_')[0]

        impath =  '/home/caffe/data/Ch/images/'+video+'/'+sp[0] +'.jpg' 

        img = cv2.imread(impath)

        resized_img = resize_image(img, (255,255))

        oversampledImages = oversample([resized_img], (224,224)) #5 crops x 2 mirror flips = return 10 images

        transposed_img = np.zeros( (10, 3, 224, 224), dtype='f4' ) 
        tp =  np.zeros( (1, 3, 224, 224), dtype='f4' ) 

        predictedLabels = []

        for j in range(0,oversampledImages.shape[0]-1):
            transposed_img[j] = oversampledImages[j].transpose((2,0,1))


            tp[0] = transposed_img[j]
            net.blobs['data'].data[0]  = tp

            pred =  net.forward(data=tp)

            predictedLabels.append(pred['prob'].argmax())


        print(predictedLabels)

        prediction,num_most_common = Cnt(predictedLabels).most_common(1)[0]


        print(prediction)

        readImgCounter = readImgCounter + 1

        if (prediction == int(sp[1])):
            runningCorrect = runningCorrect + 1

        runningAcc = runningCorrect / readImgCounter
        print('runningAcc:')
        print(runningAcc)
        print('-----------')
        print('runningCorrect:')
        print(runningCorrect)
        print('-----------')
        print('totalImgRead:')
        print(readImgCounter)
        print('-----------')

        testResultsFile.write(sp[0]+' '+str(prediction)+' '+sp[1])
        testResultsFile.write('\n')

1 Ответ

0 голосов
/ 09 октября 2018

Я исправил эту проблему в конце концов.Я не уверен на 100%, что сработало, но, скорее всего, оно меняло смещение на 0 во время обучения.

...