--augment agrument does not work in predict.py file.

[Vlad188-1]

Hello! Thank you for your repository. I have a question for you. I trained yolov7 segmentation om my custom dataset and I want to do inference for the video.
This command works fine:
python3 segment/predict.py --weights "/home/user/Disk/Priroda/git_reps/yolov7-segmentation/runs/train-seg/yolov7-seg18/weights/best.pt" --source "/home/user/Disk/Whales/From_Sasha_3/20_01/crop1_DJI_0481.mp4" --imgsz 640

When I want to slightly increase quality of inference I specify --augment parameter but it does not work.

Report error looks like this:

[RizwanMunawar]

@Vlad188-1! it seems the error is related to the None type, Let me test the issue in detail and then I will let you know or maybe directly able to fix it.

[Vlad188-1]

@RizwanMunawar Ok. Thank you!

[RizwanMunawar]

@Vlad188-1 The save-txt module is added for segmentation. You can use the mentioned command below to get coordinates of segmented area.
python3 segment/predict.py --weights yolov7-seg.pt --source "videopath.mp4" --save-txt

For --augment argument issue, I will resolve as soon as possible

[Vlad188-1]

@RizwanMunawar Thank you for your work!