搜索
您的当前位置:首页正文

分类网络中的数据标注操作

来源:二三娱乐
#给训练集标注数据
for i in range(cerebellum_label.shape[0]):
#index是cerebellum所在层的索引
    if i in index:
        y = [1,0]            
    else:
        y = [0,1]
    y_train.append(y)
print(len(y_train))
y_train = np.array(y_train)

训练完成后,我们用训练好的模型对一套没有进行过标注的数据进行预测

#model为训练网络保存的模型
label_test = model.predict(imgs, batch_size=24, verbose=1)
label_test = np.where(label_test>=0.5,1,0)
index = []
for i, label in enumerate(label_test):
    if label[0]==1:
        if label[1]==0:
            #现在的index就是预测的z轴索引范围
            index.append(i)
Top