search

xavihart / Diff-PGD

[NeurIPS'2023] Official Code Repo:Diffusion-Based Adversarial Sample Generation for Improved Stealthiness and Controllability
80 stars 6 forks source link

How to generate an accurate mask corresponding to the image #7

Closed Raineasy closed 8 months ago

Raineasy commented 9 months ago

Nice and interesting code! But how you generate an accurate mask corresponding to the image. Did you use code processing or some other tool like Photoshop? I don't see the part that generates the mask in your code, they seem to be part of the preprocessed ImageNet dataset. Sorry to bother you again. 👍

xichenpan commented 8 months ago

You can use grouned-SAM to generate good masks (https://github.com/IDEA-Research/Grounded-Segment-Anything).