Add cloth category selection feature to u2net_cloth_seg

[szriru]

Overview

This Pull Request adds cloth category selection feature when using u2net_cloth_seg

Changes Made

Users now can select which cloth category to "rembg" with a command -cc or --cloth-category. The values are "upper" or "lower" or "full", so it can be used like: rembg i input.png out.png --model u2net_cloth_seg --cloth-category lower

Mensioned Issue

484

Testing

Tried "i" and "p" command with the new option "-cc" and "--cloth-category", and it worked fine.

Result example

python ./rembg.py i ./cc-example.jpg ./cc-example-out.jpg -om --model u2net_cloth_seg --cloth-category upper

Input

Output

Reviewer Notes

I would appreciate your review of this Pull Request.

Thank you.

[danielgatis]

Wow! Thank you so much. It would be nice to use the existing extra option instead of creating another option.

for example:

rembg i input.png out.png -m u2net_cloth_seg -x '{ "cloth-category": "lower" }'

[szriru]

@danielgatis Thank you for reviewing. I commited the fixed code. It can be used with args like you provided: -x '{ "cloth-category": "lower" } or --extras '{ "cc": "upper" } I would appreciate if you can review again.

I tested with these commands, and it worked fine.

python rembg.py i input.jpg output.jpg -m u2net_cloth_seg -om python rembg.py i input.jpg output.jpg -m u2net_cloth_seg -om -x '{"cc": "lower"}' python rembg.py i input.jpg output.jpg -m u2net_cloth_seg -om --extras '{"cloth_category": "upper"}'

((We have to be careful because it will not work correctly if we swap single and double quotes due to a problem in cmd or shell or in the argment parser we use. We should put like '{"cc": "lower"}' not "{'cc': 'lower'}"))