java cnn图像识别_CNN回归预测每个测试图像的值相同 .

博主正在使用VGG-16网络对64x64x3 RGB图像进行回归预测,目标是预测图像补丁的粗糙度(范围0-160)。尽管训练了网络,但在测试时所有图像都被预测为同一值。训练损失高于验证损失,尝试过更换优化器和批量大小,但问题未解决。网络结构包括多个卷积层和池化层,最后的全连接层用于回归。
摘要由CSDN通过智能技术生成

所以只是设置我想解决的问题 . 我有大约200k 64x64x3 RGB图像的机器人开过来的地形补丁 . 每个补丁都有一个相应的标签,表示该图像补丁的粗糙度 . 粗糙度值范围为0-160 . 在机器人以不同的速度驱动时收集数据,因此粗糙度值的范围 . 我的目标是能够预测贴片的粗糙度 . 我正在使用VGG-16网络,最后一层被修改为进行回归 . 我的批量大小是1024,损失是平均sqaured错误,优化是rmsprop . 网络如下所示 . 我的问题是,在训练之后,网络预测每个测试图像的完全相同的值 . 另一点需要注意的是,训练损失总是高于奇数的验证损失 . 最后,我尝试了其他优化器,如SGD和Adam,以及不同的批量大小 . 现在我正在尝试从头开始训练网络,但这似乎不太有希望 . 我不确定这里出了什么问题,我真的很感激我能得到的任何帮助 . 谢谢

if input_tensor is None:

img_input = Input(shape=input_shape)

else:

if not K.is_keras_tensor(input_tensor):

img_input = Input(tensor=input_tensor, shape=input_shape)

else:

img_input = input_tensor

# Block 1

x = Conv2D(64, (3, 3), activation='relu', padding='same', name='block1_conv1')(img_input)

x = Conv2D(64, (3, 3), activation=&#

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值