SamsungLabs / ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation
MIT License
637 stars 125 forks source link

Using prior masks #29

Closed joshmyersdean closed 2 years ago

joshmyersdean commented 2 years ago

Hello!

Thank you for the great work. I was wondering if you could:

  1. Point me to where in your code you utilize prior masks, I'm having trouble locating it.
  2. If it's not obvious in the code describe how you go about using past model predictions? Do you precompute these or are they thresholded and saved?

Thank you!

joshmyersdean commented 2 years ago

Was able to find it in this file https://github.com/saic-vul/ritm_interactive_segmentation/blob/708182de6e3a73da4034579935aef4c19ceae80a/isegm/engine/trainer.py#L277, thank you!