训练集:场景1:10月16日发布的训练数据。场景2:10月20日发布的训练数据。
验证集:同上,因此是封闭域训练。
graph TD;
Resnet1d --> Bi-LSTM;
Bi-LSTM --> MLP1;
Bi-LSTM --> MLP2;
MLP1 --> BCE_Loss;
MLP2 --> MSE_Loss;
BCE_Loss -.-> A(predict 0/1);
MSE_Loss -.-> B(predict number of person);
A --> Output;
B --> Output;
见/predict/
目录,每日更新。
训练参数见/config/config.yaml
。由于该任务模型相对简单、稳定,模型参数直接写死在代码里了,见/model/net.py
。