whai362 / PSENet

Official Pytorch implementations of PSENet.
Apache License 2.0
1.17k stars 346 forks source link

如何将S1,S2,...Sn的结果显示到图像上,而不是只显示Sn,新手小白,求指点。 #98

Closed Donaghys closed 3 years ago

RoseSakurai commented 3 years ago

从outputs中分离出S1,S2,...Sn. outputs[bs,1, img_size(0), img_size(1)], outputs[bs,2, img_size(0), img_size(1)], outputs[bs,3, img_size(0), img_size(1)]...