数据增强的神器,非它俩莫属。albumentations靠着 OpenCV 加持,增强图像又快又稳,像Blur
、VerticalFlip
这些常见操作,几行代码就搞定,性能也不错,训练集大了不止一点点。
而Augmentor就蛮适合刚上手的朋友了,操作方式更直观,支持可视化,还能带边界框的图像,做目标检测也没压力。嗯,两者搭配用效果更好,一套组合拳直接提升模型泛化能力。
你要是搞图像分类、目标检测这类任务,真建议收藏下这份使用手册。里面步骤清晰,例子也接地气,照着撸就能跑。
顺带推荐几个相关资源,像图像分类方法、CIFAR-10 资源这些,都挺适合配合用的。
如果你在训练模型时数据不够多,或者想提升模型表现力,可以试试这两套库,轻松搞定数据增强。