How to do few-shot segmentation in 5-shot setting using BAM with my image files

[takeshiho0531]

Thank you very much for sharing your amazing work!

I am a complete beginner in computer vision.

I would like to do few-shot segmentation in 5-shot setting using BAM with my image data.
I have 5 jpg image files prepared by myself and their annotated png files as Support images and one Query image jpg file.
From here, I want to get the image data similar to the images shown in the BAM line of Visualization section in README.md of this repository.

However, I have no idea what code to write to achieve what I described above. Could you tell me what code to write?

[chunbolang]

Sorry for the late reply.

You should attempt to modify the test.py function for visualization and the get_item function in dataset.py for reading your private images.

Best,