pytorch3 YOLOv5 PyTorch 돌려보기 (작성 중) https://github.com/ultralytics/yolov5 GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Contribute to ultralytics/yolov5 development by creating an account on GitHub. github.com 위가 원작자 코드인가보다 STAR가 가장 많다. https://github.com/ultralytics/yolov5/wiki/Docker-Quickstart GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > Core.. 2023. 1. 18. TensorRT 활용해보기 (Pytorch example) TensorRT를 막상 써보려니 뭔가 잘 정리되어 있는 내용들이 없어서 따로 업데이트 해보려고 한다. 우선 내가 자주 쓰는 framework는 pytorch지만 tensorflow에서 또한 사용을 해야하기 때문에 두 케이스 모두에 대해서 알아보려고 한다. [pytorch]는 TRT [tensorflow]는 TF-TRT 로 나뉜다고 한다 (pytorch에 대해서 먼저 tensorRT를 만들고 tensorflow를 나중에 적용해서 앞에만 TF가 붙은건가??) Documentation은 아래와 같은 듯 보인다. [TRT] https://developer.nvidia.com/tensorrt#what-is NVIDIA TensorRT An SDK with an optimizer for high-performanc.. 2022. 7. 25. TorchVision model funetuning Pytorch에서 fine-tuning하는 방법을 확인해보려고 합니다. 모델은 torchvision models 중 선택할 수 있으며, 모두 1000-class Imagenet datasets로 pre-trained되었습니다. 참고 링크에 fine-tuning과 feature-extraction 이렇게 두 가지 타입의 transfer learning을 수행합니다. 1) Fintuning : pre-trained된 모델로 시작하여 새로운 task에 대한 model의 모든 parameter를 업데이트합니다. 본질적으로 전체 model을 retraining 합니다. 2) Feature extraction : pre-trained된 모델로 시작하여 prediction을 도출하는 마지막 레이어의 weight만 업.. 2021. 1. 30. 이전 1 다음
