search

Wuziyi616 / SlotDiffusion

Code release for NeurIPS 2023 paper SlotDiffusion: Object-centric Learning with Diffusion Models
https://slotdiffusion.github.io/
MIT License
79 stars 8 forks source link

Segmentation on personal dataset #2

Closed titusadt closed 1 year ago

titusadt commented 1 year ago

Hi, your work looks great, and I am tryin to use it to segment my own images. and I am just unsure of how I would go about that. I was thinking of using the pretrained weights and CLEVRTex, what confuses me is figuring out where to insert my input data. Any suggestions would be helpful, thank you

Wuziyi616 commented 1 year ago

Hi thanks for your interest in our work. This is the file we use for evaluating image segmentation. Maybe replace the input image with your images.