语义分割数据增强python代码

                                                       语义分割数据增强python-pytorch代码-语义分割github项目

0. 先放github开源项目链接

https://github.com/gengyanlei/segmentation_pytorch

 

包含网络：deeplab_v3、deeplab_v3+、unet、pspnet、HED、HF_FCN、U^2Net。。。不断更新中，将本人写的其它博客合并至此！

1. 语义分割数据增强操作

(1)随机翻转(上下-左右)

(2)随机旋转(任意角度)

(3)随机放缩(随机放大-裁剪， 随机缩小-加黑边)

(3)随机高斯噪声

(4)随机高斯模糊

(5)随机颜色(hsv等操作)

(6)随机仿射变换(参考yolov5)

(7)mosaic ??  CopyPaste??

2. 语义分割数据增强python代码

注：此数据增强可用于pytorch和tensorflow，但是本人仅将其适配于pytorch

输入图像通道>=3，标签图通道=1

(1)基于PIL+torchvision.functional组合形式对RGB-3通道的图像进行数据处理

代码链接：https://github.com/gengyanlei/segmentation_pytorch/blob/master/utils/aug_PIL.py

(2)基于opencv+GDAL读取的组合形式对3及3以上通道的图像进行数据处理

代码链接：https://github.com/gengyanlei/segmentation_pytorch/blob/master/utils/aug_GDAL.py

3.语义分割项目-pytorch版

开头的链接，此项目已经基本完成，当前版本为v1版本，后续继续更新修复bug，并优化dist，形成v2版本。