人工智能实验评价指标_GitHub - AIOpenData/baai-federated-learning-baseline-metric: 电力人工智能数据竞赛指标计算方法和自动评测脚本的详细介绍...

电力人工智能数据竞赛自动评测

由于本次电力人工智能数据竞赛采用了基于国网电力真实工业场景的指标，因此我们给出该指标计算方法和自动评测脚本的详细介绍。

环境要求

python==3.7.6

terminaltables==3.1.0

torch==1.6.0

tqdm==4.42.1

项目结构

.

├── README.md

├── crane_evaluation

│   ├── main.py # 吊车评测主函数

│   ├── test.json # 真实标注吊车json文件(未公开)

│   ├── test_pred_simple.json # 选手提交预测吊车json文件

│   └── utils # 评测功能函数

│   ├── options.py

│   └── utils.py

├── helmet_evaluation

│   ├── main.py # 安全帽评测主函数

│   ├── test.json # 真实标注安全帽json文件(未公开)

│   ├── test_pred_simple.json # 选手提交预测安全帽json文件

│   ├── utils # 评测功能函数

│   │   ├── options.py

│   │   └── utils.py

└── requirements.txt # 需要安装的python库

注：由于test.json为测试集标注答案，因此不能公开。

运行方式

计算给定预测结果和真实结果的国网电力指标并且返回：