segmentation_pytorch icon indicating copy to clipboard operation
segmentation_pytorch copied to clipboard

改了一大堆小问题,用voc数据集2个类型报类别错误

Open fenghuoxiguozu opened this issue 2 years ago • 2 comments

image f5a0c2838c2cd1b57c1dc8c9d2c6f38 loss回传时突然从tensor转成地址了

fenghuoxiguozu avatar Jul 07 '22 03:07 fenghuoxiguozu

@fenghuoxiguozu sorry,才看到,一直没有登录。 这个仓库代码是没有经过测试的,所以可能会有很多问题。如果你基于voc训练,建议你从github上搜1个在voc上训练的star比较高,效果比较好的模型;或者去paper with code上查看在voc数据集上表现优异的模型及其代码链接。 我这里主要是想实现数据增强,但是由于工作原因,未完成测试,仅提供数据增强的一些idea。

gengyanlei avatar Aug 08 '22 01:08 gengyanlei

另外voc里面有255这种边界框,每个区域的边界都是这种,有专门处理voc的,就是根据[N,C,H,W]把边界框过滤掉在loss计算时变成[N,X]的形式计算,我这里没有考虑他的这种情况。

gengyanlei avatar Aug 08 '22 01:08 gengyanlei