med-air / Endo-FM

[MICCAI'23] Foundation Model for Endoscopy Video Analysis via Large-scale Self-supervised Pre-train
Apache License 2.0
146 stars 14 forks source link

Asking about SUN dataset #8

Closed ZhangT-tech closed 2 months ago

ZhangT-tech commented 6 months ago

Hi, I went through the sun.py, I see it only process the SUN-Negative dataset. I wonder if you also train the SUN-Positive for the Endo-FM foundation model, or only the Negative cases?

Kyfafyd commented 6 months ago

Hi, thanks for your interest! We also use positive samples, as you can see in sun_seg.py.