YOLOv3学习笔记之数据处理

这篇博客详细介绍了YOLOv3模型在数据处理方面的步骤,包括图片的reshape为正方形,以及标签数据的处理。针对从json文件中获取的原始坐标,通过转换得到中心点坐标和宽高的相对值。同时,还提到了使用labelImg打标工具的数据处理方式,以及在处理batch数据时如何添加索引。此外,文章还讨论了多尺度训练中图片resize的策略和batch的构建方法。
摘要由CSDN通过智能技术生成

YOLOv3学习笔记之数据处理

目录

 

dataset中的数据处理

1. 图片处理

将读取的图片进行reshape,变为正方形

def pad_to_square(img, pad_value):
    c, h, w = img.shape
    dim_diff = np.abs(h - w)
    # (upper / left) padding and (lower / right) padding
    pad1, pad2 = dim_diff // 2, dim_diff - dim_diff // 2
    # Determine padding
    pad = (0, 0, pad1, pad2) if h <= w else (pad1, pad2, 0, 0)
    #(左填充,右填充,上填充,下填充)
    # Add padding
    img = F.pad(img, pad, "constant"
  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值