본문 바로가기

728x90
반응형
SMALL

개발

(19)
YOLOv5: Export.py # YOLOv5 🚀 by Ultralytics, AGPL-3.0 license """ Export a YOLOv5 PyTorch model to other formats. TensorFlow exports authored by https://github.com/zldrobit Format | `export.py --include` | Model --- | --- | --- PyTorch | - | yolov5s.pt TorchScript | `torchscript` | yolov5s.torchscript ONNX | `onnx` | yolov5s.onnx OpenVINO | `openvino` | yolov5s_openvino_model/ TensorRT | `engine` | yolov5s.engine..
YOLOv5 갖고 놀기 python benchmarks.py --weights yolov5s.pt --img 640 라는 명령어를 돌리니 대충 돌아가는 것 처럼 보인다. 그러나 내가 진짜 원하는 동작을 수행해야 한다. 1. 내가 원하는 학습을 수행해야 한다. 2. 학습의 결과가 제대로 전시되어야한다. 3. tflite 모델로 출력이 나와야한다.(현재는 XX.pt) 확장자로 나오는것 같다. 최초로 예제 파일을 받아보면, 여러가지 파일이 있고 대체 뭐부터 봐야 할지 눈이 뒤집힐 지경이다. 배경지식 없이 개발을 하면 이런 문제가 간혹있는데, 어자피 퍼스트무버가 아니니 다른 사람의 레퍼런스를 참조해서 시간을 절약해보자. 1. 내가 원하는 학습을 수행하자. 이는 커스텀트레이닝(custom Training)이라고 하며, 하는 방식으로는 ..
YOLOv5 처음 실행해보기 OLOv5를 돌려보자, 환경구축은 이전글을 참조하면된다. YOLOv5 개발환경 구축하기 어찌됬던 YOLOv8모델로 작업을 진행하려 했으나 TFLITE 변환에 실패하고 말았다. YOLO 공식사이트에서도 TFLITE 변경을 지원하는 정식 버전을 YOLOv5로 안내하는 것으로 보아 v5가 모바일에 탑재하기 가 ipsmaster.tistory.com 일단 냅다 https://docs.ultralytics.com/yolov5/tutorials/model_export/ 사이트 들어가서 내용을 시키는 대로 진행해보자. 그리고 시키는데로 벤치마크를 한번 돌려보자. python benchmarks.py --weights yolov5s.pt --imgsz 640 --device 0 당연하게도 에러가 발생하면서 동작을 하지..
YOLOv5 개발환경 구축하기 어찌됬던 YOLOv8모델로 작업을 진행하려 했으나 TFLITE 변환에 실패하고 말았다. YOLO 공식사이트에서도 TFLITE 변경을 지원하는 정식 버전을 YOLOv5로 안내하는 것으로 보아 v5가 모바일에 탑재하기 가장 적합하다는 판단을 했다. 완전 초보자도 함께 진행 할 수 있도록 아주 쉽게 설명해놓도록 하겠다. 먼저 https://github.com/ultralytics/yolov5 사이트에 방문하여 다운로드를 진행하자. 파이썬에서 가상환경을 만들고 난 뒤 git clone을 통해 소스코드를 받기로 하겠다. 아무 폴더나 만들고 여기서 커맨드창을 실행시킨다. 위에 cmd 만 입력하고 엔터를 치면 커맨드창이 생성된다. python -m venv yolov5 라고 입력한다. 가상환경을 설정하는 이유는 오래..
YOLOv5 to tflite https://docs.ultralytics.com/yolov5/tutorials/model_export/#exported-model-usage-examples
플러터참조 https://medium.flutterdevs.com/live-object-detection-app-with-flutter-and-tensorflow-lite-a6e7f7af3b07 Live Object Detection App With Flutter and TensorFlow Lite Let’s learn how to build a flutter app that detects objects on a live camera. medium.flutterdevs.com ain.dart: 앱의 메인 파일입니다. MyApp 위젯을 생성하고, runApp 함수를 통해 앱을 실행합니다. MyApp: StatelessWidget으로, 앱의 홈 화면을 구성하는 HomePage 위젯을 반환합니다. HomePage: St..
YOLOv8 to tflite YOLOv8을 학습하여 tflite 파일로 모델을 추출하고 싶은데 에러가 계속 발생한다. PyTorch: starting from 'runs\detect\train9\weights\best.pt' with input shape (1, 3, 640, 640) BCHW and output shape(s) (1, 84, 8400) (6.2 MB) TensorFlow SavedModel: export failure ❌ 3.6s: Descriptors cannot be created directly. If this call came from a _pb2.py file, your generated code is out of date and must be regenerated with protoc >= 3.19.0...
YOLOv8 맛보기 지난 며칠동안 삽질을 해가면서 YOLOv8을 돌려보았다. https://docs.ultralytics.com/ko/ 홈 Ultralytics YOLOv8을 완벽하게 탐구하는 가이드로, 고속 및 정확성이 특징인 객체 탐지 및 이미지 분할 모델입니다. 설치, 예측, 훈련 튜토리얼 등이 포함되어 있습니다. docs.ultralytics.com 간단하게 공홈에 들려서 욜로의 역사와 이것저것 알아보았다. 여러 버전이 있지만 큰 고민 없이 바로 v8 버전으로 작업을 수행하기로 결정했다. 라이센스는 기업체가 사용할 경우에는 라이센스 비용을 지불해야 하니 주의하도록 하자. 난 학습 용도로 쓰니 문제 없을 것이고, 나중에 회사에서 적용할 때는 다른 무료 라이센스를 찾던지 아니면 비용을 지불하는 방향으로 진행해야겠다. 일..

728x90
반응형
LIST