Я публикую здесь свою проблему, поскольку у меня нет доступа к группе Google Caffe.
У меня есть два класса изображений, которые я хочу классифицировать с помощью caffe.Эти изображения имеют 4 канала и были преобразованы в lmdb для фазы обучения.
Обучение прошло хорошо, однако, когда модель тестируется с изображениями png с 4 каналами, классификатор оценивает всегда один и тот же класс.
Заранее спасибо
Я пытался адаптировать преобразователь для чтения 4-канальных изображений, но я не знаю, правильно ли я это сделал
image = cv2.imread(imgstest_folder,cv2.IMREAD_UNCHANGED).astype(np.float32)/255.0
transformer.set_transpose('data', (2,0,1))
transformer.set_channel_swap('data', (3,2,1,0))
transformed_image = transformer.preprocess('data', image)
net.blobs['data'].data[i] = transformed_image