poseC3D代码结构

说明:虽然代码看着很多,其实只有三部分:1. 配置文件   2. 工具包     3. 训练器

一. 配置文件--config

说明网络的框架,文件名以主干网络为主

1. C3D    2. slowonly  3. x3d

下图以C3D中的C3D_light_ntu60为例

b0ec07198f404c6cbe1f7be156638521.png

说明:ann_file 下载在 tool/data/data_doc中查找链接

二. 工具包--pyskl

1. models: 被 config/joint.py 调用

2. dataset: 

    (1) base.py   pose_dataset.py  video_dataset.py:  说明数据集的类型,是把视频转化为图片,还是直接对视频进行处理

 (2)datasets/pipeline: 数据预处理部分

 (3)datasets/data_wrappers.py: 重复数据集的长度,个人理解为成倍数的扩大训练的轮数,因为 在joint.py中 total_epoches=24,在data=dic(time=10),官方提供的model zoo中写道训练240轮次

说明:只要在config/joint.py中调用的,都有build.py  因为joint是以字典的形式,在调用时要把调用的类放到容器中 (个人理解)

3. 剩下的 apis  core  utils在tools/train.py调用

三. 训练器--tool/train.py or test.py

1. 调用pyskl/apis/train.py

2. 在apis/train.py中的数据加载和模型加载中都使用了分布式训练

(1)数据加载器的相关设置中,dataloader_setting=dict(video_per_gpu等),调用了build_dataloader

(2)模型加载中,model=MMDistributedDataParallel()

 

 

 

  • 4
    点赞
  • 25
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值