Cannot run the file run_one_image.py

YihanHu-2022 / DiffMatte

61 stars 4 forks source link

Cannot run the file run_one_image.py #6

Closed thangnguyenmd closed 1 week ago

thangnguyenmd commented 2 months ago

The command I used to tets Diff Matte:

python run_one_image.py --model vitmatte-s --checkpoint-dir ./DiffMatte_ViTS_Com_1024.pth --image-dir 80209689_889049583_525_1_f_mask_0_person_6.png --trimap-dir 80209689_889049583_525_1_f_mask_0_person_6.png --output-dir ./test_matting

but get the error: run_one_image.py: error: unrecognized arguments: --model

can you help me to fix the issue?

YihanHu-2022 commented 1 month ago

sorry for the late response, it seems that you are using the wrong command to run the demo code. We have fixed this command in run_one_image.py, hope this can help you. :)