计算机视觉
super_gradients 标签描述

问题 在使用super_gradients库中的Yolo-nas预测图片时,想要获取预测好的图片,但执行out=model.predict("camera01.png",conf=0.6,batch_size=None)之后只能out.show()和out.save(),无法返回预测结果图片。 解决方法 importcv2 importtorch fromsuper_gradients.trainingimportmodels device=torch.device("cuda:0")iftorch.cuda.is_available()elsetorch.device("cpu") yo...