COCOデータセットから特定のデータを抽出し、それと別のデータを合わせてYOLOv4で物体認識させたい

Question

具体的には、YOLOv4の環境構築をした時に付いてあったCOCOデータから「car」「motorbike」「track」「bus」の車両関係のクラスだけを抽出し、それとは別の物体も合わせてYOLOv4に物体認識させたいです。
つまり、COCOデータの車両関係とCOCOデータにはない別の物体だけをYOLOv4に物体認識させるようにしたいです。

現状は、YOLOv4やdarknet関連の環境構築は終わっており、「yolov4.weights」「yolov4.cfg」「coco.data」の3つのファイルを使って「darknet.exe detector demo cfg/coco.data cfg/yolov4.cfg yolov4.weights xxx.mp4」と実行すると80個のクラスで物体認識できる状態になっています。
この80個のクラスの中から4個の車両関係のクラスと、80個のクラスにはない別のクラス(自前のもの)だけを物体認識できるようにしたいです。

これを実現するにはどのファイルなどをどのように編集すればよいのでしょうか？

Answer 1

できますよ。
あたりまえですが、全部自前のデータでもOkですし。

Comments

[neg315]

ご回答ありがとうございます。
具体的には、どのファイルをどう編集すれば良いのでしょうか？

[TakaiY]

それを知りたいのであれば、質問を編集したほうがいいですね。
また、現状何ができているのかなどの情報がないと、1から全部説明というのは難しいと思います。

[neg315]

わかりました。ご対応ありがとうございます。