AttaNet模型的PyTorch版本复现 配置文件 config.yaml: 参数设置 数据集的准备 文件夹格式 ├─data_dir │ ├─images │ │ └─train │ └─labels │ └─train 训练模型 python main.py 评价系数 python evaluate.py 导出模型为ONNX python export.py 推理 python inference.py