I3D Finetune UCF101

上个月主要是基于TensorFlow使用4卡训练,用Kinetics400数据集预训练的I3D模型,对UCF101 split1进行finetune,从训练精度和效率上做了很多零碎的实验,稍微总结一下实验结果:

1、首先只使用RGB帧,基本上训几个epoch就能达到95.7%,这个问题不大,学习率合适即可。四卡大概两三小时就训出来了。

2、投了一些时间在TensorFlow的多机多卡的加速上,由于公司都是使用集群,所以真实的端口和IP尤其重要,注意区分容器的端口和IP。另外表示使用dataset API的话,速度比队列快,然后多进程,预取都要用上,尽可能把内存小的操作放在前面速度更快,速度太慢可以考虑cache。用htop观察CPU,同时watch nvidia-smi观察GPU实时占用率,看看是不是IO卡训练速度,当然也可以使用前面说的timeline。

3、tf.train.MonitoredTrainingSession()是个好东西啊,写代码更简洁了。

4、实现了快速的video IO(即读入video然后on the fly的做decoding)!!!!这个很重要!!!在大数据集下(如Kinetics400或者Kinetics600)on the fly decode比先把video转成frames再来处理会高效而且方便很多,不需要先解码视频存成帧在硬盘。Kinetics400视频大概130+G,处理成RGB帧估计要1T以上,不仅浪费存储而且预处理总是费时间的。我也在不断优化tf代码,现在on the fly的训练速度基本上和解码存帧差不多甚至更快。关键发现在于opencv > videoFileclip, nvvl>= ffmpeg ,经同事提醒才知道opencv解码真心快。虽然在后面三个也分别实现了,甚至考虑了cache,浪费了一点时间。

5、使用video IO 训UCF101,当然也是更快的复现了95.7%的精度咯~

至此,I3D与UCF101算是告一段落了。

总结完,想到今年在三星中国研究院打activitynet18的时候,当时使用的还是自己改的caffe,然后预取RGB和optical Flow存硬盘里,才能进行训练,而且需要SSD。现在想想,当时真的应该早点学TensorFlow写video IO,然后跑3D CONV,一劳永逸。不过话说回来,以当时的资源情况。也只能把2D玩到极限,caffe的速度还是挺快的,不过由于存图太多(大概一两亿图片,现在感觉是肯定卡了IO,实际速度应该更快),存储需要太大,三星买了两个4T硬盘,现在想想觉得其实如果实现了快速video IO,本是不需要买的。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 194,761评论 5 460
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 81,953评论 2 371
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 141,998评论 0 320
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 52,248评论 1 263
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 61,130评论 4 356
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 46,145评论 1 272
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 36,550评论 3 381
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 35,236评论 0 253
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 39,510评论 1 291
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 34,601评论 2 310
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 36,376评论 1 326
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,247评论 3 313
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 37,613评论 3 299
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 28,911评论 0 17
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,191评论 1 250
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 41,532评论 2 342
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 40,739评论 2 335

推荐阅读更多精彩内容

  • 教程一:视频截图(Tutorial 01: Making Screencaps) 首先我们需要了解视频文件的一些基...
    90后的思维阅读 4,632评论 0 3
  • 1、通过CocoaPods安装项目名称项目信息 AFNetworking网络请求组件 FMDB本地数据库组件 SD...
    X先生_未知数的X阅读 15,960评论 3 119
  • 曾经有人问我你怎么去看生死,当时我想了想给他讲了一个故事! 在一次聚众讲道时,佛祖问西游四人:人生有多长? 沙僧...
    戒掉ai阅读 756评论 0 2
  • 古田四路。 四月的尾巴开始散发出毕业的味道,周六申论划完最后一个句号,一个星期的时间,可期的日子里开始了毕业论文,...
    吴少年啊阅读 570评论 0 2
  • 愿一个人好好生活,安静独处,不希望有人打扰,直到慢慢老去,身体回归泥土
    爱吃小花生阅读 194评论 0 0