“WithoutBackground”

[gaoyixuan111]

@JackAILab In the definition of the masks_for_unique_values function in your functions.py script, if the parsing_mask passed in does not contain any pixels with value=0, then the generated parsing_mask_list will not have the key "WithoutBackground". This means that in the __getitem__ method, you cannot generate the corresponding multi_clip_image. Could you add a solution for this issue?

body_raw_image = fetch_mask_raw_image(raw_image,parsing_mask_list["WithoutBackground"]) body_clip_image = self.clip_image_processor(images=body_raw_image, return_tensors="pt").pixel_values multi_clip_image = torch.cat([clip_image, body_clip_image], dim=1)