らんだむな記憶

blogというものを体験してみようか!的なー

YOLOv3 (1)

darknet の比較 - らんだむな記憶 で触れた

darknet は読みたくない、PyTorch だけで読みたいって場合には https://github.com/eriklindernoren/PyTorch-YOLOv3 とか https://github.com/ultralytics/yolov3 を見たら良さそう。

だが、後者を見るとリリースページに用意されている .pt などをロードしている。一方前者は darknet の訓練済みウェイトをダウンロードしてきて models.py で darknet 用の .cfg ファイルをパースしながら PyTorch の言葉に置き換えてモデルを構築している。この意味では驚くほどストレートであると感じるのでまずはこちらから眺めてみたい。