我有下面的代码,它获取一组图像,每个训练集中大约50个图像,然后创建一个线性模型并尝试对数据进行分类。我也有一个测试集,但它甚至不能对训练数据进行任何准确的分类。我加载图像的方式有什么错误吗?如果有帮助的话,我很乐意提供更多的代码或输出。在
def create_image_list(file_path):
image_list = []
for filename in glob.glob(file_path):
img = Image.open(filename)
img_resized = img.resize((32, 32), Image.ANTIALIAS)
pix = img.load()
pixlist = []
for x in range(0, 32):
for y in range(0,32):
pixlist.append(pix[x,y][0])
pixlist.append(pix[x,y][1])
pixlist.append(pix[x,y][2])
image_list.append(pixlist)
return image_list
dalmation_training = create_image_list('/images/dalmatian/training/*')
dollabill_training = create_image_list('/images/dollar_bill/training/*')
pizza_training = create_image_list('/images/pizza/training/*')
soccer_ball_training = create_image_list('/images/soccer_ball/training/*')
sunflower_training = create_image_list('/images/sunflower/training/*')
c = '1e2'
testing_set = dalmation_training + dollabill_training + pizza_training + soccer_ball_training + sunflower_training
dalmation_y = [1]*len(dalmation_training ) + [-1]*len(dollabill_training) + [-1]*len(pizza_training) + [-1]*len(soccer_ball_training) + [-1]*len(sunflower_training)
dalmation_model_linear = svm_train(dalmation_y, testing_set, '-t 0 -c %s -b 1 -q' % c)
dollabill_y = [-1]*len(dalmation_training ) + [1]*len(dollabill_training) + [-1]*len(pizza_training) + [-1]*len(soccer_ball_training) + [-1]*len(sunflower_training)
dollabill_model_linear = svm_train(dollabill_y, testing_set, "-t 0 -c %s -b 1 -q" % c)
pizza_y = [-1]*len(dalmation_training ) + [-1]*len(dollabill_training) + [1]*len(pizza_training) + [-1]*len(soccer_ball_training) + [-1]*len(sunflower_training)
pizza_model_linear = svm_train(pizza_y, testing_set, "-t 0 -c %s -b 1 -q" % c)
soccer_ball_y = [-1]*len(dalmation_training ) + [-1]*len(dollabill_training) + [-1]*len(pizza_training) + [1]*len(soccer_ball_training) + [-1]*len(sunflower_training)
soccer_ball_model_linear = svm_train(soccer_ball_y, testing_set, "-t 0 -c %s -b 1 -q" % c)
sunflower_y = [-1]*len(dalmation_training) + [-1]*len(dollabill_training) + [-1]*len(pizza_training) + [-1]*len(soccer_ball_training) + [1]*len(sunflower_training)
sunflower_model_linear = svm_train(sunflower_y, testing_set, "-t 0 -c %s -b 1 -q" % c)
print 'dalmation linear'
result1, something, p1 = svm_predict([1]*len(testing_set), testing_set, dalmation_model_linear, "-b 1")
print 'dollabill linear'
result2, something, p2 = svm_predict([1]*len(testing_set), testing_set, dollabill_model_linear, "-b 1")
print 'pizza linear'
result3, something, p3 = svm_predict([1]*len(testing_set), testing_set, pizza_model_linear, "-b 1")
print 'soccer linear'
result4, something, p4 = svm_predict([1]*len(testing_set), testing_set, soccer_ball_model_linear, "-b 1")
print 'sunflower linear'
result5, something, p5 = svm_predict([1]*len(testing_set), testing_set, sunflower_model_linear, "-b 1")
当我运行这个数据集并运行一些精度测量时,每次使用最后一个数据集时,它大约是20%,向日葵的准确率接近100%,其他的接近5%。我相信我把它放在libsvm的正确格式中,我找不到任何线索。我试过从1e-8到1e8的不同c值,每一个的准确度变化不超过5%。在
任何意见都将非常感谢,我很乐意提供更多的信息!在
testing_set
列表传递给svm_predict
,对于真正的标签,您传递{dalmation_y
。在相关问题 更多 >
编程相关推荐