yolo-nas使用教程
创建yolo-nas环境
conda create -n yolonas python=3.8
# 安装yolo-nas包
pip install super-gradients
导入包并设置模型参数
import torch
from super_gradients.training import models
device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu")
# 使用预训练的参数
model = models.get("yolo_nas_s", pretrained_weights="coco").to(device)
模型推理
out = model.predict("test_1.jpg", conf=0.6)