Current situation
The coco dataset example seem to highlight how unaligned is the dataset item and the function to visualize it :
It requires
- image conditionning
- label conditionning
- boxes conditionning
- parsing of each of the 3 to the drawing function
suggestion
in order to make the COCO dataset more accessible, we could also manage the coco dataset item as a S3 class that would be directly managed by draw_bounding_boxes() and draw_segmentation_mask() . This would turn the example into the very straigthforward
coco <- coco_detection_dataset( root = "~/data", train = FALSE, year = "2017", download = TRUE)
item <- coco[1]
segmented_item <- draw_segmentation_mask(item)
tensor_image_browse(segmented_item)
@koshtiakanksha, what is your view on this ?