discopat.nn_training.torch_detection_utils.coco_utils

Functions

`convert_coco_poly_to_mask`(segmentations, ...)
`convert_to_coco_api`(ds)
`get_coco`(root, image_set, transforms[, ...])
`get_coco_api_from_dataset`(dataset)

Classes

`CocoDetection`(img_folder, ann_file, transforms)
`ConvertCocoPolysToMask`()

class discopat.nn_training.torch_detection_utils.coco_utils.CocoDetection(img_folder, ann_file, transforms)[source]: Bases: CocoDetection

class discopat.nn_training.torch_detection_utils.coco_utils.ConvertCocoPolysToMask[source]: Bases: object

discopat.nn_training.torch_detection_utils.coco_utils.convert_coco_poly_to_mask(segmentations, height, width)[source]

discopat.nn_training.torch_detection_utils.coco_utils.convert_to_coco_api(ds)[source]

discopat.nn_training.torch_detection_utils.coco_utils.get_coco(root, image_set, transforms, mode='instances', use_v2=False, with_masks=False)[source]

discopat.nn_training.torch_detection_utils.coco_utils.get_coco_api_from_dataset(dataset)[source]