我是靠谱客的博主 靓丽小蝴蝶,最近开发中收集的这篇文章主要介绍poseC3D代码结构一. 配置文件--config二. 工具包--pyskl三. 训练器--tool/train.py or test.py,觉得挺不错的,现在分享给大家,希望可以做个参考。

概述

说明:虽然代码看着很多,其实只有三部分: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()

 

 

 

最后

以上就是靓丽小蝴蝶为你收集整理的poseC3D代码结构一. 配置文件--config二. 工具包--pyskl三. 训练器--tool/train.py or test.py的全部内容,希望文章能够帮你解决poseC3D代码结构一. 配置文件--config二. 工具包--pyskl三. 训练器--tool/train.py or test.py所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错,欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(46)

评论列表共有 0 条评论

立即
投稿
返回
顶部