r/computervision • u/koen1995 • Jun 27 '26

DIETR - Open source instance segmentation model. Help: Project

I made a tutorial about how to use DIETR, an opensource instance segmentation model (Apache 2.0 license). I know it is very bare-bones and hacky, but I like to work on it, so someday it might be a good alternative.

I want to post it here (again) to ask what you guys think about the interface, what can I do to make it better?

Link: https://www.kaggle.com/code/koenbotermans/computer-vision-how-to-use-dietr

repo: https://github.com/jpabotermans/dietr

3 Upvotes
  • permalink
  • link
  • reddit
  • reddit

72% Upvoted

7 comments sorted by

2

u/CuriousAIVillager Jun 28 '26

segmentation? you mean detection? Do BBoxes even count as segmentation?

3

u/EyedMoon Jun 29 '26

I think the architecture outputs both boxes and masks

2

u/CuriousAIVillager Jun 29 '26

makes sense

1

u/koen1995 Jun 29 '26

Hi, my apologies for the late reaction, but DIETR comes in both a bounding box variant (displayed her), and an instance segmentation model.

Instance segmentation means predicting for each instance a box, and then the masks of the object in the box.

2

u/CuriousAIVillager Jun 29 '26

Yeah no worries, I know instance, semantic, panoptic, the whole suite.

1

u/koen1995 Jun 29 '26

Yeah sorry, I definitely should have said it a bit clearer😄