caffe图像分类教程_Caffe - 多级和多标签图像分类

只是试图理解实用的方法..在创建包含图像的所有标签的2个.text文件(一个用于训练，一个用于验证)之后，例如：

/train/img/1.png 0 4 18

/train/img/2.png 1 7 17 33

/train/img/3.png 0 4 17

运行py脚本：

import h5py, os

import caffe

import numpy as np

SIZE = 227 # fixed size to all images

with open( 'train.txt', 'r' ) as T :

lines = T.readlines()

# If you do not have enough memory split data into

# multiple batches and generate multiple separate h5 files

X = np.zeros( (len(lines), 3, SIZE, SIZE), dtype='f4' )

y = np.zeros( (len(lines),1), dtype='f4' )

for i,l in enumerate(lines):

sp = l.split(' ')

img = caffe.io.load_image( sp[0] )

img = caffe.io.resize( img, (SIZE, SIZE, 3) ) # resize to fixed size

# you may apply other input transformations h