关于引入黑边的问题

[xiaobai-marker]

您好，我做的是二值的语义分割，0为背景（黑色），1为标签（白色），您在图像增强脚本里说对于旋转，偏移等操作，会引入黑边(均为0值)，将引入的黑边 视为1类，标签值默认为0，真实标签从1开始，引入的黑边为1类，导致我分割出的图全为白色，请问这个问题怎么修改呢

[gengyanlei]

@xiaobai-marker 背景作为1类也参与训练，没问题的啊。你用的我的代码么？