Real-Time Detection
-
[YOLO] YOLOv5-custom 학습(신호등 인지)YOLO 2023. 6. 21. 16:12
이번 포스트는 기존 YOLOv3-Darknet 사용법 이후로 YOLOv5(Pytorch)로 버전을 바꾸기 위해 테스트를 진행했습니다. *현재 작성자 기준 세팅 1. Nvidia driver - 470 2. Cuda - 11.2 3. CuDnn - 8.1.1 4. Pytorch - 1.8.0 + cu111 !!주의!!) 위 세팅 기준으로 작성하였기에 사용자 분들은 현재 GPU와 기본세팅에 맞춰서 준비하셔야합니다. 사용법 참고) https://docs.ultralytics.com/yolov5/tutorials/train_custom_data/#before-you-start Train Custom Data Train your custom dataset with YOLOv5. Learn to collect, l..