[pytorch] 图像识别之mixup/cutmix

最新推荐文章于 2024-08-29 08:00:50 发布

MachineLP

最新推荐文章于 2024-08-29 08:00:50 发布

阅读量3.1k

点赞数 2

分类专栏： Deep learning 比赛方案总结文章标签：图像识别 mixup cutmix

本文链接：https://blog.csdn.net/u014365862/article/details/104216086

版权

比赛方案总结同时被 2 个专栏收录

11 篇文章 91 订阅 ¥39.90 ¥99.00

订阅专栏

Deep learning

175 篇文章 13 订阅

订阅专栏

本文介绍了如何使用PyTorch实现图像识别中的mixup和cutmix技术，通过分享的Kaggle讨论链接展示了这两种技术在目标检测任务中的应用效果。

摘要由CSDN通过智能技术生成

本人kaggle分享链接：https://www.kaggle.com/c/bengaliai-cv19/discussion/126504

效果图：（目标检测中）

代码如下：



def rand_bbox(size, lam):
    W = size[2]
    H = size[3]
    cut_rat = np.sqrt(1. - lam)
    cut_w = np.int(W * cut_rat)
    cut_h = np.int(H * cut_rat)

    # uniform
    cx = np.random.randint(W)
    cy = np.random.randint(H)

    bbx1 = np.clip(cx - cut_w // 2, 0, W)
    bby1 = np.clip(cy - cut_h // 2, 0, H)
    bbx2 = np.clip(

了解本专栏