카테고리 없음
[진저비어] 하계 모각코 2주차 회고
1. window에서 YOLO model -> tensorRT 세팅하기from ultralytics import YOLOimport torchdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")print(torch.cuda.is_available())# 모델 로드model = YOLO("./trained_yolov8/best.pt")model.to(device)# PyTorch to TensorRTmodel.export(format='engine', device=device, half=True)변환할 모델을 load한 뒤, export함수를 통해 engine 포맷으로 변환 시켜주면 된다.이 때 주의할 점은 cuda를 사용할 수 있는 ..
