shanemankiw / Panodiff

Official implementation for ACM MM 2023 paper '360-Degree Panorama Generation from Few Unregistered NFoV Images'
44 stars 2 forks source link

mask generation #7

Open sp77-1 opened 2 months ago

sp77-1 commented 2 months ago

Thank you for sharing your wonderful work! how can i get mask of input images in my custom dataset?

shanemankiw commented 1 month ago

Hi, sorry for the late reply.

If you are talking about how to generate a crop mask for the panorama, you can refer to this: https://github.com/shanemankiw/Panodiff/blob/b6e86122150b43f660c3ea49cc503574a7ef6140/scripts/dataset/crop_generator.py#L68

Given theta and phi angles, one can use this to generate the corresponding mask of the patch.