YOLO模型使用所遇到的问题

YOLOv3

问题1:已解决

已知:YOLOv3训练的图片是(416,416,3)size大小
测试单张图片时,对一张(1600,1200,3)分辨率大小的图片进行测试,分别以原图和原图resize为(416,416,3)进行测试,发现用原图测试效果更好,why?
原图

resize后
解决:所用yolo代码作者先将图片进行BiCubic插值方法进行resize为一个尺寸,再将图片用image.paste填充到一个(416,416,3)大小的像素全为(128,128,128)图片上来进行测试,之后再反馈到原图。
代码:

'''resize image with unchanged aspect ratio using padding'''
# image为输入图片,size为目标尺寸即(416,416,3)
def letterbox_image(image, size):
    iw, ih = image.size
    w, h = size
    scale = min(w/iw, h/ih)
    nw = int(iw*scale)
    nh = int(ih*scale)

    image = image.resize((nw,nh), Image.BICUBIC)
    new_image = Image.new('RGB', size, (128, 128, 128))
    new_image.paste(image, ((w-nw)//2, (h-nh)//2))

    return new_image
问题2:已解决

训练时最终loss很低,检测效果却很差,检测不到目标,调试发现训练图片转为的tensor全为0。
解决:将图片的box信息转为0-1比时,定义的矩阵没有给定float类型,导致转换结果全为0.

问题3:未解决

将yolo网络的输出,计算所有格子的class类别,发现实际检测时预测目标的概率很低
解决

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 4
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值