import pickle


def load_data(input_path):
    """ Load coco dataset """
    with open(input_path, 'rb') as file:
        coco_raw = pickle.load(file)
    image_categories = coco_raw['image_categories']
    image_file = coco_raw['image_file']
    category_id = coco_raw['category_id']

    return image_categories, image_file, category_id