C3D-tensorflow

GitHub
585 260 较难 1 次阅读 7个月前MIT视频开发框架
AI 解读 由 AI 自动生成,仅供参考

C3D-tensorflow 是一个基于 TensorFlow 框架实现的开源项目,旨在复现经典的 C3D(3D 卷积神经网络)模型。它主要解决了视频动作识别领域的技术落地问题,让开发者能够直接在 TensorFlow 环境中利用成熟的 C3D 架构进行视频特征提取与分类训练,而无需依赖原始的 Caffe 框架。

该工具特别适合人工智能研究人员、深度学习工程师以及对视频分析感兴趣的技术开发者使用。其核心亮点在于支持直接加载由原版 C3D-Caffe 转换而来的预训练模型,大幅降低了迁移门槛。项目内置了针对 UCF101 数据集的完整处理脚本,涵盖从视频解码、帧提取到训练列表生成的全流程,并提供了训练、验证及随机片段测试的代码示例。

值得注意的是,为了兼容不同来源的预训练权重,C3D-tensorflow 在数据维度变换上做了细致处理,用户可根据加载的模型类型灵活调整转置操作。虽然在 UCF101 数据集上的视频级准确率与原 caffe 版本存在约 5% 的细微差距,但它依然为希望在 TensorFlow 生态中探索 3D 卷积网络的用户提供了一个高效、实用的基准实现方案。

使用场景

某安防科技公司的算法团队正致力于开发一套基于监控视频的智能行为识别系统,需要准确判断画面中人员是否发生了打架、跌倒或奔跑等特定动作。

没有 C3D-tensorflow 时

  • 时空特征提取困难:传统 2D 卷积网络只能逐帧分析图像,无法有效捕捉动作在时间维度上的连续变化规律,导致对快速动作的误判率极高。
  • 框架迁移成本高昂:团队主要技术栈为 TensorFlow,但业界成熟的 C3D 预训练模型多基于 Caffe 框架,手动重写网络结构并转换权重耗时且极易出错。
  • 数据预处理繁琐:缺乏标准化的视频解码与切片流程,开发人员需自行编写脚本将视频按固定帧率(如 5FPS)拆解并重排,难以复现论文中的"16 帧重叠切片”策略。
  • 模型精度难以对齐:由于缺少经过验证的 TensorFlow 实现参考,自研模型在 UCF101 数据集上的视频级准确率始终比官方基准低 10% 以上,无法满足交付标准。

使用 C3D-tensorflow 后

  • 原生支持 3D 卷积:直接利用修改版的 TensorFlow 架构运行 3D ConvNets,天然具备同时提取空间外观与时间运动特征的能力,显著提升了对复杂动态行为的识别精度。
  • 无缝复用预训练权重:工具支持直接加载从 Caffe 转换而来的 sports1m_finetuning_ucf101.model 等成熟模型,省去了从头训练的漫长周期,实现了“开箱即用”。
  • 内置标准化数据处理:提供 convert_video_to_images.sh 等脚本,一键完成视频解码、帧抽取及训练列表生成,确保输入数据格式严格符合 C3D 论文的切片规范。
  • 性能表现可预期:在 UCF101 数据集上复现了接近原版的视频级准确率(仅约 5% 的边际差距),让团队能快速基于可靠的基线模型进行业务场景的微调与优化。

C3D-tensorflow 通过打通框架壁垒并固化最佳实践,让开发者能在熟悉的 TensorFlow 环境中高效构建高精度的视频动作识别应用。

运行环境要求

操作系统
  • 未说明
GPU

未说明 (基于 TensorFlow 1.2+,通常建议配备支持 CUDA 的 NVIDIA GPU 以加速 3D 卷积训练)

内存

未说明

依赖
notes1. 必须下载 UCF101 动作识别数据集。2. 视频文件需预先解码为图像序列(脚本默认按 5FPS 提取),并生成对应的 train.list 和 test.list 索引文件。3. 加载部分预训练模型(如 sports1m_finetuning_ucf101.model)时,代码中必须包含特定的转置操作 (transpose perm=[0,1,4,2,3]),而加载其他模型则需注释掉该行。4. 该项目是 C3D-caffe 的 TensorFlow 实现,在 UCF101 split1 上的准确率比原版 Caffe 实现低约 5%。
python未说明 (需兼容 TensorFlow >= 1.2 的版本,推测为 Python 2.7 或 3.5-3.6)
tensorflow>=1.2
Pillow
C3D-tensorflow hero image

快速开始

C3D-tensorflow

这是一个尝试在 TensorFlow 上实现 C3D-caffe 的仓库,直接使用从原始 C3D-caffe 转换而来的模型。请注意,我们在 TensorFlow 中的实现与原始 C3D-caffe 在 UCF101 split1 上的视频级别准确率相差约 5%。

需求:

  1. 已安装 TensorFlow >= 1.2 版本
  2. 必须安装以下两个 Python 库: a) tensorflow b) Pillow
  3. 必须下载 UCF101(动作识别数据集)
  4. 每个单独的 AVI 文件以 5 FPS 的帧率解码(具体取决于您的选择),并存储在一个目录中。
    • 您可以使用 ./list/convert_video_to_images.sh 脚本来解码 UCF101 视频文件
    • 运行 ./list/convert_video_to_images.sh .../UCF101 5
  5. list 目录下生成 {train,test}.list 文件。每行对应一个“图像目录”和一个类别(从 0 开始)。例如:
    • 您可以使用 ./list/convert_images_to_list.sh 脚本来为数据集生成 {train,test}.list
    • 运行 ./list/convert_images_to_list.sh .../dataset_images 4,这将在根文件夹内按比例 4 生成 test.listtrain.list 文件。
database/ucf101/train/ApplyEyeMakeup/v_ApplyEyeMakeup_g01_c01 0
database/ucf101/train/ApplyEyeMakeup/v_ApplyEyeMakeup_g01_c02 0
database/ucf101/train/ApplyEyeMakeup/v_ApplyEyeMakeup_g01_c03 0
database/ucf101/train/ApplyLipstick/v_ApplyLipstick_g01_c01 1
database/ucf101/train/ApplyLipstick/v_ApplyLipstick_g01_c02 1
database/ucf101/train/ApplyLipstick/v_ApplyLipstick_g01_c03 1
database/ucf101/train/Archery/v_Archery_g01_c01 2
database/ucf101/train/Archery/v_Archery_g01_c02 2
database/ucf101/train/Archery/v_Archery_g01_c03 2
database/ucf101/train/Archery/v_Archery_g01_c04 2
database/ucf101/train/BabyCrawling/v_BabyCrawling_g01_c01 3
database/ucf101/train/BabyCrawling/v_BabyCrawling_g01_c02 3
database/ucf101/train/BabyCrawling/v_BabyCrawling_g01_c03 3
database/ucf101/train/BabyCrawling/v_BabyCrawling_g01_c04 3
database/ucf101/train/BalanceBeam/v_BalanceBeam_g01_c01 4
database/ucf101/train/BalanceBeam/v_BalanceBeam_g01_c02 4
database/ucf101/train/BalanceBeam/v_BalanceBeam_g01_c03 4
database/ucf101/train/BalanceBeam/v_BalanceBeam_g01_c04 4
...

使用方法

  1. python train_c3d_ucf101.py 将训练 C3D 模型。训练好的模型将保存在 models 目录中。
  2. python predict_c3d_ucf101.py 将在验证数据集上测试 C3D 模型。
  3. cd ./C3D-tensorflow-1.0 && python Random_clip_valid.py 将使用提供的 sports1m_finetuning_ucf101.model 在 UCF101 测试集上获得随机片段的准确率。
  4. C3D-tensorflow-1.0/Random_clip_valid.py 代码兼容 TensorFlow 1.0 及以上版本,与旧仓库略有不同。
  5. 重要提示:当您加载 sports1m_finetuning_ucf101.model 时,应使用转置操作,例如:pool5 = tf.transpose(pool5, perm=[0,1,4,2,3]),或者在 Random_clip_valid.py 中写成:["transpose", [0, 1, 4, 2, 3]]。但如果您加载的是 conv3d_deepnetA_sport1m_iter_1900000_TF.modelc3d_ucf101_finetune_whole_iter_20000_TF.model,则不需要进行转置操作,只需注释掉该行代码即可。

实验结果:

  • 注意:

    1. 所有报告的结果均基于 UCF101 split1(训练视频:9537,测试视频:3783)。
    2. 所有结果均为视频级别的准确率,除非另有说明。
    3. 我们采用与 C3D 论文相同的视频片段提取方式:“为了提取 C3D 特征,将视频分割为长度为 16 帧的片段,相邻片段之间有 8 帧的重叠。这些片段被输入到 C3D 网络中以提取 fc6 激活值。然后对这些片段的 fc6 激活值取平均,形成一个 4096 维的视频描述符,并对其进行 L2 归一化。”
  • C3D 作为特征提取器:

平台 特征提取器模型 fc6+SVM fc6+SVM+L2 归一化
caffe conv3d_deepnetA_sport1m_iter_1900000.caffemodel 81.99% 83.39%
tensorflow conv3d_deepnetA_sport1m_iter_1900000_TF.model 79.38% 81.44%
tensorflow c3d_ucf101_finetune_whole_iter_20000_TF.model 79.67% 81.33%
tensorflow sports1m_finetuning_ucf101.model 82.73% 85.35%
  • 在 UCF101 split1 上使用预训练模型对 C3D 网络进行微调:
平台 预训练模型 训练策略 视频准确率 片段准确率 随机片段
caffe c3d_ucf101_finetune_whole_iter_20000.caffemodel 直接测试 - 79.87% -
tensorflow c3d_ucf101_finetune_whole_iter_20000_TF.model 直接测试 78.35% 72.77% 57.15%
tensorflow-A conv3d_deepnetA_sport1m_iter_1900000_TF.caffemodel 整体微调 76.0% 71% 69.8%
tensorflow-B sports1m_finetuning_ucf101.model 冻结卷积层,仅微调全连接层 79.93% 74.65% 76.6%
  • 注:
    1. tensorflow-A 模型对应于 @ hx173149 提供的在 UCF101 上预训练的原始 C3D 模型。
    2. tensorflow-B 模型是在 tensorflow-A 的基础上冻结了卷积层,并以 learning rate=1e-3 对全连接层进行了额外四轮微调。
    3. “随机片段”列表示从 UCF101 测试集 split1 中的每个视频中随机选择一个片段,因此结果并不十分稳健。但根据大数定律,我们可以假设该项目与您的视频级别准确率呈正相关。
    4. 显而易见的是,如果您基于 c3d_ucf101_finetune_whole_iter_20000_TF.model 进行更多的微调工作,可能会取得更好的效果,但由于时间限制,我没有这样做。
    5. 毫无疑问,通过适当微调网络,您可以获得更好的结果。

训练好的模型:

模型 描述 云存储 下载
C3D sports1M TF C3D sports1M,由 Caffe 版本的 C3D 转换而来 Dropbox C3D sports1M
C3D UCF101 TF C3D UCF101 训练模型,由 Caffe 版本的 C3D 转换而来 Dropbox C3D UCF101
C3D UCF101 TF 训练 在 UCF101 split1 数据集上进行微调,使用 @ hx173149 提供的 C3D sports1M 模型 Dropbox C3D UCF101 split1
split1 均值文件 TF UCF101 split1 的均值文件,由 Caffe 版本的 C3D 转换而来 Dropbox UCF101 split1 均值文件
上述所有文件 上述四个文件 百度网盘 百度网盘

其他版本

参考资料:

常见问题

相似工具推荐

openclaw

OpenClaw 是一款专为个人打造的本地化 AI 助手,旨在让你在自己的设备上拥有完全可控的智能伙伴。它打破了传统 AI 助手局限于特定网页或应用的束缚,能够直接接入你日常使用的各类通讯渠道,包括微信、WhatsApp、Telegram、Discord、iMessage 等数十种平台。无论你在哪个聊天软件中发送消息,OpenClaw 都能即时响应,甚至支持在 macOS、iOS 和 Android 设备上进行语音交互,并提供实时的画布渲染功能供你操控。 这款工具主要解决了用户对数据隐私、响应速度以及“始终在线”体验的需求。通过将 AI 部署在本地,用户无需依赖云端服务即可享受快速、私密的智能辅助,真正实现了“你的数据,你做主”。其独特的技术亮点在于强大的网关架构,将控制平面与核心助手分离,确保跨平台通信的流畅性与扩展性。 OpenClaw 非常适合希望构建个性化工作流的技术爱好者、开发者,以及注重隐私保护且不愿被单一生态绑定的普通用户。只要具备基础的终端操作能力(支持 macOS、Linux 及 Windows WSL2),即可通过简单的命令行引导完成部署。如果你渴望拥有一个懂你

349.3k|★★★☆☆|1周前
Agent开发框架图像

stable-diffusion-webui

stable-diffusion-webui 是一个基于 Gradio 构建的网页版操作界面,旨在让用户能够轻松地在本地运行和使用强大的 Stable Diffusion 图像生成模型。它解决了原始模型依赖命令行、操作门槛高且功能分散的痛点,将复杂的 AI 绘图流程整合进一个直观易用的图形化平台。 无论是希望快速上手的普通创作者、需要精细控制画面细节的设计师,还是想要深入探索模型潜力的开发者与研究人员,都能从中获益。其核心亮点在于极高的功能丰富度:不仅支持文生图、图生图、局部重绘(Inpainting)和外绘(Outpainting)等基础模式,还独创了注意力机制调整、提示词矩阵、负向提示词以及“高清修复”等高级功能。此外,它内置了 GFPGAN 和 CodeFormer 等人脸修复工具,支持多种神经网络放大算法,并允许用户通过插件系统无限扩展能力。即使是显存有限的设备,stable-diffusion-webui 也提供了相应的优化选项,让高质量的 AI 艺术创作变得触手可及。

162.1k|★★★☆☆|1周前
开发框架图像Agent

everything-claude-code

everything-claude-code 是一套专为 AI 编程助手(如 Claude Code、Codex、Cursor 等)打造的高性能优化系统。它不仅仅是一组配置文件,而是一个经过长期实战打磨的完整框架,旨在解决 AI 代理在实际开发中面临的效率低下、记忆丢失、安全隐患及缺乏持续学习能力等核心痛点。 通过引入技能模块化、直觉增强、记忆持久化机制以及内置的安全扫描功能,everything-claude-code 能显著提升 AI 在复杂任务中的表现,帮助开发者构建更稳定、更智能的生产级 AI 代理。其独特的“研究优先”开发理念和针对 Token 消耗的优化策略,使得模型响应更快、成本更低,同时有效防御潜在的攻击向量。 这套工具特别适合软件开发者、AI 研究人员以及希望深度定制 AI 工作流的技术团队使用。无论您是在构建大型代码库,还是需要 AI 协助进行安全审计与自动化测试,everything-claude-code 都能提供强大的底层支持。作为一个曾荣获 Anthropic 黑客大奖的开源项目,它融合了多语言支持与丰富的实战钩子(hooks),让 AI 真正成长为懂上

160.4k|★★☆☆☆|今天
开发框架Agent语言模型

ComfyUI

ComfyUI 是一款功能强大且高度模块化的视觉 AI 引擎,专为设计和执行复杂的 Stable Diffusion 图像生成流程而打造。它摒弃了传统的代码编写模式,采用直观的节点式流程图界面,让用户通过连接不同的功能模块即可构建个性化的生成管线。 这一设计巧妙解决了高级 AI 绘图工作流配置复杂、灵活性不足的痛点。用户无需具备编程背景,也能自由组合模型、调整参数并实时预览效果,轻松实现从基础文生图到多步骤高清修复等各类复杂任务。ComfyUI 拥有极佳的兼容性,不仅支持 Windows、macOS 和 Linux 全平台,还广泛适配 NVIDIA、AMD、Intel 及苹果 Silicon 等多种硬件架构,并率先支持 SDXL、Flux、SD3 等前沿模型。 无论是希望深入探索算法潜力的研究人员和开发者,还是追求极致创作自由度的设计师与资深 AI 绘画爱好者,ComfyUI 都能提供强大的支持。其独特的模块化架构允许社区不断扩展新功能,使其成为当前最灵活、生态最丰富的开源扩散模型工具之一,帮助用户将创意高效转化为现实。

109.2k|★★☆☆☆|今天
开发框架图像Agent

gemini-cli

gemini-cli 是一款由谷歌推出的开源 AI 命令行工具,它将强大的 Gemini 大模型能力直接集成到用户的终端环境中。对于习惯在命令行工作的开发者而言,它提供了一条从输入提示词到获取模型响应的最短路径,无需切换窗口即可享受智能辅助。 这款工具主要解决了开发过程中频繁上下文切换的痛点,让用户能在熟悉的终端界面内直接完成代码理解、生成、调试以及自动化运维任务。无论是查询大型代码库、根据草图生成应用,还是执行复杂的 Git 操作,gemini-cli 都能通过自然语言指令高效处理。 它特别适合广大软件工程师、DevOps 人员及技术研究人员使用。其核心亮点包括支持高达 100 万 token 的超长上下文窗口,具备出色的逻辑推理能力;内置 Google 搜索、文件操作及 Shell 命令执行等实用工具;更独特的是,它支持 MCP(模型上下文协议),允许用户灵活扩展自定义集成,连接如图像生成等外部能力。此外,个人谷歌账号即可享受免费的额度支持,且项目基于 Apache 2.0 协议完全开源,是提升终端工作效率的理想助手。

100.8k|★★☆☆☆|1周前
插件Agent图像

markitdown

MarkItDown 是一款由微软 AutoGen 团队打造的轻量级 Python 工具,专为将各类文件高效转换为 Markdown 格式而设计。它支持 PDF、Word、Excel、PPT、图片(含 OCR)、音频(含语音转录)、HTML 乃至 YouTube 链接等多种格式的解析,能够精准提取文档中的标题、列表、表格和链接等关键结构信息。 在人工智能应用日益普及的今天,大语言模型(LLM)虽擅长处理文本,却难以直接读取复杂的二进制办公文档。MarkItDown 恰好解决了这一痛点,它将非结构化或半结构化的文件转化为模型“原生理解”且 Token 效率极高的 Markdown 格式,成为连接本地文件与 AI 分析 pipeline 的理想桥梁。此外,它还提供了 MCP(模型上下文协议)服务器,可无缝集成到 Claude Desktop 等 LLM 应用中。 这款工具特别适合开发者、数据科学家及 AI 研究人员使用,尤其是那些需要构建文档检索增强生成(RAG)系统、进行批量文本分析或希望让 AI 助手直接“阅读”本地文件的用户。虽然生成的内容也具备一定可读性,但其核心优势在于为机器

93.4k|★★☆☆☆|1周前
插件开发框架