https://github.com/likelyzhao/MXNet_Video_Trainer

simple training system for video training
https://github.com/likelyzhao/MXNet_Video_Trainer

mxnet video

Last synced: 2 months ago
JSON representation

simple training system for video training

• Host: GitHub
• URL: https://github.com/likelyzhao/MXNet_Video_Trainer
• Owner: likelyzhao
• Created: 2017-11-02T05:41:58.000Z (about 7 years ago)
• Default Branch: master
• Last Pushed: 2017-11-07T10:55:48.000Z (about 7 years ago)
• Last Synced: 2024-11-03T21:42:48.269Z (2 months ago)
• Topics: mxnet, video
• Language: Python
• Size: 44.9 KB
• Stars: 6
• Watchers: 2
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• Awesome-MXNet - MXNet_Video_Trainer

README

        
# MXNet_Video_Trainer

simple training system for video training 

## 模型框图

![](resource/structure.png)

包含 主控模块(Master)，数据提供模块(data provider) 和 MXNet 训练模块 以及若干个消息队列。

### 主控模块：

主控模块负责后面两个模块的创建和销毁，同时负责向input的消息队列里面传入截帧，计算光流和截取一系列帧所需要的信息，同时收取训练结束的反馈。

### 数据提供模块：

前面接一个输入参数的消息队列，并完成对应的截帧，计算光流或者是截取一系列帧后，将数据和label写入训练数据的消息队列

### MXNet 训练模块：

基于MXNet的DataIter ，实现一个输入一系列消息队列的数据输入layer。

定义对应的symbol文件，设置超参，启动训练

## 实现：

1. 单机实现。

2. 队列使用python多线程队列，

3. 截帧，计算光流等用ffmpeg 直接搞定，

4. 需要开发一个读取特定队列的mxnet dataloader

5. 联机调试进行训练

## 多线程队列测试通过

调用 `python train.py` 查看测试效果

`config.py` 中 `config.RUNTIMECONFIG.num_provider_thread = 10` 可以设置 `providier` 的线程数目