ModuleNotFoundError: No module named 'pc2label'

[Rumi07]

Dear Authors,

Thank you so much for your outstanding work. While I was trying to run the inference.py file I got the following error:
ModuleNotFoundError: No module named 'pc2label'

Would you please share the pc2label file? I did not find this file in the code.

FairDiff/PointMaskGen/inference.py

Line 17 in 71ab24c

from pc2label import *

I really appreciate any help you can provide.

[wenyi-li]

We have updated the code, please check the latest commit.
Let us know if you run into any other issues, and we'd be happy to help!

[Rumi07]

Dear Author,

Thank you so much, now it works.
Would you please share from where I can get this 'control_sd15_seg.pth' model weights?

FairDiff/MaskImageGen/train.py

Line 19 in e1d4a90

resume_path = './models/control_sd15_seg.pth'

Thanks again, I'm looking forward to hearing back from you.

[wenyi-li]

We utilize ControlNet to generate images from masks.
You can find model weights here:
https://huggingface.co/lllyasviel/ControlNet/blob/main/models/control_sd15_seg.pth