How to modify the points simulator?

SamsungLabs / ritm_interactive_segmentation

Reviving Iterative Training with Mask Guidance for Interactive Segmentation

MIT License

637 stars 125 forks source link

How to modify the points simulator? #12

Open nickhong92509 opened 3 years ago

nickhong92509 commented 3 years ago

If I want to change the strategy of points simulation, which part of code should I modify? I don't know how points_"sampler.py" and get_next_points in "trainer.py" working, whick one is in charge of simulation. Thanks!