Some images are stuck in DBSCAN

[xiaoyao9184]

watermark-anything/notebooks/inference_utils.py

Line 150 in 45d56c2

db = DBSCAN(eps=epsilon, min_samples=min_samples).fit(valid_pred)

Conducting multi-region scans on single-region watermarked images can be quite slow, which is understandable because separation is not possible.

However, some images seem to cause the DBSCAN algorithm to get stuck indefinitely. like this

Since I can't understand why the DBSCAN algorithm is getting stuck, I vaguely noticed that high-resolution images seem not to have this issue. Perhaps I should adjust the epsilon and min_samples parameters based on the image size like this scikit-learn/scikit-learn#17650 ?

[pierrefdz]

Hi, sorry for the delay of answer.

• Would you be able to send your code?
• I have the same issue when trying to extract one message from more than 40% of image pixels. I think it comes from the fact that DBSCAN clustering must run on too many pixels. The issue is more complicated than I thought and may be tricky to solve. I'll try to have a look at it in the following days. If you find something please tell me :)