dddb11/MVSS-Net

运行问题 'mask' is not defined

CongYep opened this issue · 1 comments

我使用没有边缘mask的数据进行训练,txt的格式为Manipulated image without pre-generated edge mask,运行代码提示错误:
/datasets/dataset.py", line 342, in init
self.edge_generator((mask>0.5) * 1.0)[0][0]
NameError: name 'mask' is not defined

Sorry for this bug, I have fixed it in the latest commit. Please check it.
如果这个问题解决了,你可以close或者进一步讨论,目前的实现还是有些慢,五一过后我会改成运行速度更快的版本。