[HELP] DataBlock API for Instance Segmentation

How can i create a DataBlock for the mask of a BBoxBlock?

This is useful for making instance segmentation for example with RetinaMask o MaskRCNN or other SOTA models.

I saw how to make object detection in this notebook.

He uses something like the following:

pascal = DataBlock(blocks=(ImageBlock, BBoxBlock, BBoxLblBlock),
                 splitter=RandomSplitter(),
                 get_items=get_train_imgs, 
                 getters=getters,
                 item_tfms=item_tfms,
                 batch_tfms=batch_tfms,
                 n_inp=1)

Is enough adding MaskBlock at the end? Show_result will print well the result?

For well I mean:

1 Like