【Python】yolov5で物体検出をしてみる
こちらの記事を参考にしました。こちらの記事では方法など省略しています。
テスラ車が前のトラックが載せている信号に反応してたって話題になりましたね。
【悲報】テスラ車さん、センサーが優秀すぎてトラックに積まれた信号機を誤検知してしまう pic.twitter.com/mNAw8qHXyL
— 滝沢ガレソ🐯 (@takigare3) 2021年9月4日
満月を黄色信号と誤検知したりするので無免許で車に乗れるのはまだ遠いかもしれないです。
https://pytorch.org/hub/ultralytics_yolov5/
この記事では、使うモデルと画像を入力すると物体検出をして結果を出力するっぽいです。
物体検出した画像はローカルに保存されます。
この画像は帽子をフリスビーと誤検出していますがフリスビーも検出してくれるのかって驚きました笑
追記:コード載せました。
import torch # Model model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True) # Images imgs = ['https://cdn.pixabay.com/photo/2017/05/13/12/40/fashion-2309519_960_720.jpg'] # batch of images # Inference results = model(imgs) # Results results.print() results.save() # or .show() results.xyxy[0] # img1 predictions (tensor) results.pandas().xyxy[0] # img1 predictions (pandas)