def _register_dataset()

in source/train.py [0:0]

• 8 lines of code
• 1 McCabe index (conditional complexity)

def _register_dataset(dataset_name, annotation_file, image_dir):
    from detectron2.data.datasets import register_coco_instances

    dataset_location = os.environ["SM_CHANNEL_TRAINING"]

    register_coco_instances(dataset_name, {}, os.path.join(dataset_location, annotation_file), 
                            os.path.join(dataset_location, image_dir))

    drone_meta = MetadataCatalog.get(dataset_name)
    logger.info(f"Registered dataset {dataset_name}")
    logger.info(drone_meta)