YOLOv5の検出画像をすべて取得したい

Question

実現したいこと

val.pyでtestデータの検証をした際に、検出画像(labels、pred)がbatch2までしか取得できませんでした。すべてのtestデータで検出画像が出力された場合はbatch8まであります。どの設定を変更すれば、すべての検出画像を取得できるようになるでしょうか。

該当のソースコード

!python val.py --task test --weights runs/train/MODEL/weights/best.pt --data bccd_data_test.yaml --img 640  --batch 8  --name TEST3 --save-txt --save-conf  --conf 0.50

試したこと

上記のコードを実行ところbatch2までしか検出画像を取得できませんでした。

Answer 1

val.pyのコードでbatch_i<3の条件を外したところ取得できました。