python 機械学習プログラムのネットワーク部分が分からない。yolov5

Question

大学で機械学習を用いた画像認識の研究をしています。
プログラム中のネットワーク部分がわかりません。どなたか教えてください。
train.pyかdetect.pyのどちらかにあると思うのですが、、

追記
・外部リンクを使うのはコードが長すぎて、載せることができないため。
・コードはオープンソースなので著作権などは大丈夫

https://drive.google.com/drive/folders/1--CeVnrPvbrP9iyVGfwRaGTfRYjnb_Cd?usp=sharing

Comments

[jbpb0]

そのコードを書いた人に聞くのが早いと思いますけど

どこかから落としてきたものですか？

[tiroha]

https://medium.com/analytics-vidhya/training-a-custom-object-detection-model-with-yolo-v5-aa9974c07088
このサイトからです。
分かりました。サイトの主に質問してみます。

[jbpb0]

https://medium.com/analytics-vidhya/training-a-custom-object-detection-model-with-yolo-v5-aa9974c07088
の「Setup the YAML files for training」のところに書いてある「The second YAML is...」のあたりによると、
https://github.com/ultralytics/yolov5/tree/master/models
にいくつか置いてあるような「*.yaml」ファイルでネットワークを定義していて、それを読み込んだPythonコード内でネットワークが生成される、のではないですかね

https://github.com/ultralytics/yolov5/blob/master/models/yolo.py
の「def parse_model(...」あたりが怪しい

ちゃんと読んでないので、外してたらごめんなさい

[tiroha]

ありがとうございます！
さっそく確認してみます。

Answer 1

Training a Custom Object Detection Model With Yolo-V5
の「Setup the YAML files for training」のところに書いてある「The second YAML is...」のあたりによると、
models
にいくつか置いてあるような「*.yaml」ファイルでネットワークを定義していて、それを読み込んだPythonコード内でネットワークが生成される、のではないですかね

yolo.py
の「def parse_model(...」あたりとか