From 03fbe5df0ebe5173100b69390b3d7d035d1c24de Mon Sep 17 00:00:00 2001 From: dengkaipeng Date: Sat, 2 Mar 2019 21:05:14 +0800 Subject: [PATCH] code strcuture --- fluid/PaddleCV/video/README.md | 46 ++++++++++++++++++++++++++++++++++ 1 file changed, 46 insertions(+) diff --git a/fluid/PaddleCV/video/README.md b/fluid/PaddleCV/video/README.md index cfd96689..53ea0f2d 100644 --- a/fluid/PaddleCV/video/README.md +++ b/fluid/PaddleCV/video/README.md @@ -58,6 +58,52 @@ bash scripts/train/train_stnet.sh - 请根据`CUDA_VISIBLE_DEVICES`指定卡数修改`config`文件中的`num_gpus`和`batch_size`配置。 +## 模型库结构 + +### 代码结构 + +``` +configs/ + stnet.txt + tsn.txt + ... +dataset/ + youtube/ + kinetics/ +datareader/ + feature_readeer.py + kinetics_reader.py + ... +metrics/ + kinetics/ + youtube8m/ + muticrop_test/ + ... +models/ + stnet/ + tsn/ + ... +scripts/ + train/ + test/ +train.py +test.py +infer.py +``` + +- `configs`: 各模型配置文件模板 +- `datareader`: 提供Youtube-8M,Kinetics数据集reader +- `metrics`: Youtube-8,Kinetics数据集评估脚本 +- `models`: 各模型网络结构构建脚本 +- `scripts`: 各模型快速训练评估脚本 +- `train.py`: 一键式训练脚本,可通过指定模型名,配置文件等一键式启动训练 +- `test.py`: 一键式评估脚本,可通过指定模型名,配置文件,模型权重等一键式启动评估 +- `infer.py`: 一键式推断脚本,可通过指定模型名,配置文件,模型权重,待推断文件列表等一键式启动推断 + +### 配置文件 + +配置文件分为5段,`[MODEL]`段包含模型结构、数据集等整体配置,`[TRAIN]`,`[VALID]`,`[TEST]`,`[INFER]`段包含训练、验证、评估、推断过程中相关配置。 + ## Model Zoo - 基于Youtube-8M数据集模型: -- GitLab