from dataset import HighlightDataset






if __name__ == "__main__":

    dataset = HighlightDataset(root_dir="../data/tvsum")