datajoely
06/22/2021, 3:51 PMAbstractDataSet
and implement the __init__()
, load()
and save()
methods.
Looking at this example from pycocotools I think it is super easy for you to just need to import from pycocotools.coco import COCO
and do COCO(your_path)
within the load()
method and you're good to go.
https://github.com/cocodataset/cocoapi/blob/master/PythonAPI/pycocoDemo.ipynb
The only thing worth noting is that this will only work for local files - if you want the dataset to accept either local or cloud files (s3 etc) you may want to extend / steal the basic implementation from any of the existing Kedro datasets