from model.segformer_b2 import Segformer
from PIL import Image
from model.cloth_masker import AutoMaskerSeg
# model = Segformer("/home/chongzheng_p23/data/Projects/CatVTON-main/Models/segformer_b3_clothes")
image = Image.open("/home/chongzheng_p23/data/Projects/CatVTON-main/resource/demo/example/person/women/1-model_3.png")
# result = model(image)
# result.save("a.png")

masker = AutoMaskerSeg(
    densepose_ckpt="/home/chongzheng_p23/data/Projects/CatVTON-main/Models/densepose",
    segformer_ckpt="/home/chongzheng_p23/data/Projects/CatVTON-main/Models/segformer_b3_clothes")



result = masker(image)['mask']
result.save("b.png")