Can't reproduce visualization result

[HieuBui99]

Hello. Thanks for your great work!

I'm trying to play around with the visualization, but can't seem to reproduce the result in your example folder.

I just cloned the repository, downloaded the weight and ran the below command.

python cluster_visualize.py --image images/A.JPEG --model coc_tiny_plain --checkpoint model_best.pth.tar --stage 0 --block 0 --head 1

Result:

Compared to the example;