YOLO v10 Winform Sample 입니다.
ONNX (Open Neural Network Exchange) 이용하여 닷넷에서 작동하도록 하였습니다.
class id 는 yolo의 기본 제공하는 것입니다.
.pt를 .onnx로 변환하고 onnx를 불러와서 사용합니다.
conda init
conda create -n yolov10_env python=3.10
conda activate yolov10_env
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install ultralytics
pip install onnx
pip install opencv-python
pip install huggingface_hub
[1] .pt를 .onnx로 변환하기
export_to_onnx2.py
[2] onnx를 c# winform에서 불러와 사용하는 샘플
YoloV10OnnxSample.zip
YoloV10OnnxSample : 이미지에서 오브젝트 디텍션
YoloV10WebcamSample : 웹캠에서 오브젝트 디텍션
T4에서 TensorRT FP16으로 측정한 지연 시간 GPU.
'Dev > Deep Learning' 카테고리의 다른 글
Anomalib (1) | 2024.08.18 |
---|---|
이상 탐지 관련 오픈소스 프로젝트들 (0) | 2024.08.18 |
TensorFlow 2.x - Simple learning (0) | 2024.01.26 |
Pytorch-Simple learning (0) | 2024.01.26 |
Keras-simple deep learning (0) | 2024.01.23 |
댓글