yolov5学習について

前提

python3.8を使用しています。

実現したいこと

yolov5でカスタムデータセットを学習をしようと思っています。
そこで、画像サイズを640×480で実行しようとしているのですがうまくいきません。
ご教示よろしくお願いいたします。

発生している問題・エラーメッセージ

コマンド

python3  /YOLOv5/yolov5/train.py --img-size 640 480 --batch 8 --epochs 300 --data data.yaml --weights yolov5s.pt --cache

usage: train.py [-h] [--weights WEIGHTS] [--cfg CFG] [--data DATA] [--hyp HYP] [--epochs EPOCHS]
                [--batch-size BATCH_SIZE] [--imgsz IMGSZ] [--rect] [--resume [RESUME]] [--nosave]
                [--noval] [--noautoanchor] [--noplots] [--evolve [EVOLVE]] [--bucket BUCKET]
                [--cache [CACHE]] [--image-weights] [--device DEVICE] [--multi-scale]
                [--single-cls] [--optimizer {SGD,Adam,AdamW}] [--sync-bn] [--workers WORKERS]
                [--project PROJECT] [--name NAME] [--exist-ok] [--quad] [--cos-lr]
                [--label-smoothing LABEL_SMOOTHING] [--patience PATIENCE]
                [--freeze FREEZE [FREEZE ...]] [--save-period SAVE_PERIOD]
                [--local_rank LOCAL_RANK] [--entity ENTITY] [--upload_dataset [UPLOAD_DATASET]]
                [--bbox_interval BBOX_INTERVAL] [--artifact_alias ARTIFACT_ALIAS]
train.py: error: unrecognized arguments: 480

該当のソースコード

def parse_opt(known=False):
    parser.add_argument('--imgsz', '--img', '--img-size', type=int, default=640, help='train, val image size (pixels)')
(https://github.com/ultralytics/yolov5.git)

jbpb0

2022/11/17 00:40

https://github.com/ultralytics/yolov5/issues/516#issuecomment-671161735 に、「What should i do if i want to train my dataset resized to 640*320 ?」という質問に対して「--img 640」と回答されてるので、 > 画像サイズを640×480で実行しようとしているの場合は「--img-size 640」だけ(480は指定しない)でいいのではないですかね

jbpb0

2022/11/17 00:44

https://teratail.com/help#posted-otherservice を見てください https://stackoverflow.com/questions/74457702/image-size-during-training-in-yolov5