dlam4h / OCD-PyTorch

The code for paper 《Object Contour Detection with a Fully Convolutional Encoder-Decoder Network》
19 stars 4 forks source link

issues of test #1

Open dbAIStudio opened 5 years ago

dbAIStudio commented 5 years ago

My test result is a black picture. What is the problem?

dlam4h commented 5 years ago

@dbAIStudio 应该是没训练好。你用的是哪个代码?第一个代码有时间会出现全黑的情况。

dbAIStudio commented 5 years ago

对要训练的label中的mask图轮廓宽度有要求么,为什么我训练出模型检测时输入需要改成336大小的才可以全图训练

dlam4h commented 5 years ago

@dbAIStudio 轮廓宽度看你自己设置,不要太窄就行。输入图片尺寸的话,如果模型有4个下采样输入图片尺寸设置为16的倍数就行。不能整除的话上采样过程中的torch.cat会出现维度不匹配

dbAIStudio commented 5 years ago

3q

shanchao0906 commented 4 years ago

第一个代码出现了全黑的情况