yolov5で複数のクラス番号を指定する方法

Question

前提

yolov5を用いて，人と車のリアルタイム検知をしようと考えています．
python detect.py --source 0 --class 0
を実行することで人のリアルタイム検知はできている状況です．

実現したいこと

複数のクラス番号を指定して実行する方法はありますか？
例えば，--class0と--class2のように複数のクラスを指定したいです．人と車のみ検知するものを作りたいと考えています．
なにかいい方法があれば教えて頂きたいです．

Comments

[jbpb0]

> --class0と--class2のように複数のクラスを指定

https://github.com/ultralytics/yolov5/blob/master/detect.py#L234
の234行目を見ると、「--classes 0 2」と指定できるようです

【追記】
コメントを書いてる間に解決してました
失礼しました

Answer 1

python detect.py --source 0 --class 0 2とすれば実行できました．