今天在做眼底图像识别的时候,发现0级别和1,2,3,4级别的数量严重不等,所以想到了设计一个分级的卷积神经网络,先进行二分类,即0为一类,2,3,4为一类,这也就涉及到了训练集标签的修改了,所以用python写了一个小脚本。
train = open('trainLabels.txt','r')
train_new = open('trainLabels_new.txt','w+')
line = train.readline()
while line:
line = list(line)
label = line[len(line)-2