def METHOD_NAME(img_folder, ann_file, image_set, return_masks):
    dataset = CocoDetection(img_folder, ann_file, transforms=make_coco_transforms(image_set), return_masks=return_masks)
    return dataset