ming71 / CFC-Net

[TGRS 2021] Official implementation of "CFC-Net: A Critical Feature Capturing Network for Arbitrary-Oriented Object Detection in Remote Sensing Images".

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

代码报错,多类别的情况下

chl916185 opened this issue · comments

该行代码不适合多类别的使用的,会索引越界
cls_targets[positive_indices, assigned_annotations[positive_indices, -1].long()] = 1

该行是可以给多类别使用的,这行代码在多类识别中的正确类给1作为分类gt。可能是训练配置中类别数未更改导致和实际标签不符导致越界。