Closed li-zheng-hao closed 4 years ago
with torch.no_grad(): for x, _ in dataloaders: y=model(x) img_y=torch.squeeze(y).numpy() plt.imshow(img_y)
请问测试代码中这一段img_y输出的值代表什么呢?如何用绿色或者红色标注出图像中的目标?
请问测试代码中这一段img_y输出的值代表什么呢?如何用绿色或者红色标注出图像中的目标?