data_utils.py

import csv


def csv_to_size_list(csv_file: str) -> dict:
    # 初始化一个空字典
    size_list_dict = {}

    # 打开CSV文件并读取数据
    with open(csv_file, mode="r") as file:
        reader = csv.reader(file)
        # 跳过表头
        next(reader)
        # 读取数据并填充字典
        for row in reader:
            size_name, h, w = row
            size_list_dict[size_name] = (int(h), int(w))

    return size_list_dict