search

icey-zhang / SuperYOLO

SuperYOLO is accepted by TGRS
323 stars 54 forks source link

测试阶段输出SR结果 #84

Closed Tou-ge closed 8 months ago

Tou-ge commented 9 months ago

想在test阶段输出SR分支的结果，看测试集超分后的可视化结果，应该怎么操作

leileilei2000 commented 9 months ago

我并不确定我的回答是否正确，但是作者好像在test阶段移除了SR分支

icey-zhang commented 8 months ago

如果你想在测试阶段有SR输出，你可以在https://github.com/icey-zhang/SuperYOLO/blob/30c902c7e57d6b9658b7bcf09e03e4c050b8cb68/models/SRyolo.py#L248这之前加一句output_sr = self.model_up(y[self.l1],y[self.l2])，并且返回output_sr