YOLOPをお試ししています。
YOLOPはマルチタスクなネットワークで、車両検出、道路セグメンテーション、レーンセグメンテーションの3つのタスクをこなします🦔
YOLOPさん。
— 高橋 かずひと@マリオジェノサイダー🦔 (@KzhtTkhs) 2022年1月15日
訓練済みモデルが日本の道路と相性悪いのか、いまいちセグメンテーションが不安定な気がするんですよねー👀
マルチタスクで良いのですけども。 pic.twitter.com/bb8OShxWAA
オリジナルリポジトリは以下です。
ソースコードは以下にコミットしています。
オリジナルリポジトリのONNX推論サンプルが、
後処理でtorch tensor使ってたり、NMSにtorchvision使ってたりしてPyTorchに依存していたので、NumpyとOpenCVに置き換えています👻
