DataFountain赛题-555

DataFountain 训练赛道赛题 天气以及时间分类

---

About The Project

本仓库记录了DataFountain赛题 天气以及时间分类 的解决方案。截至2022年3月27日，测试成绩为0.94934795，第二赛段排行榜A榜排名第一。

(back to top)

Getting Started

1. 项目结构：

   • DataFountain-555
     --dataset（数据集）
     --params（模型参数）
     --source（源代码）
     ----main.py
     ----models.py
     ----train.py
     ----test.py
     ----config.py
     ----utils.py
     --voting_models（用于投票的模型）
     ----model_name1 （第一种模型）
     ------1.pt
     ------2.pt
     ----model_name2 （第二种模型）
     ------1.pt
     ------2.pt

2. 其他说明：

   • 本项目将去除了训练集中未出现的三种类别，将剩余的period中["Dawn","Morning","Afternoon","Dusk"]与weather中["Cloudy","Sunny","Rainy"]合并为3x4=12类进行分类。
   • 进行投票的模型只参与Inference过程，且具有相同权重。
   • 所有经过预训练的模型均在ImageNet上完成预训练。
   • 将参与投票的模型参数并按网络结构进行分类，放至voting_models不同文件夹下。然后于models.py中define_net建立映射关系。

(back to top)

Methods

• 迁移学习
• 数据增强
• 多模型投票
• 去除训练集中未出现的类别

(back to top)