TSN 编译记录

TSN代码编译记录

前言

TSN 作者yjxiong etc.开源了代码并且公布了自己训练的所有模型,提供了使用GPU得到tvl1光流的方法,以及几种数据集切分的方法,比自己的渣代码顺眼很多,这个项目很赞~
TSN-Project

但是(据说但是前都是废话),DL框架用的Caffe,提供的自然是caffemodel和prototxt文件,习惯了tensorflow和pyTorch,表示很不想用Caffe,而且工程里用到了opencv(必须是版本2.x),需要支持GPU。很麻烦,基本上要重新安装各个依赖。

很好的是,作者提供了build_all.sh脚本文件,一次安装完所有依赖。不建议直接在自己的服务器/主机上运行。建议使用docker。

当然,如果本机上使用的是caffe + CUDA8.0 + opencv-2.x(GPU support)可以考虑直接按照作者的思路

这大概是历史上最痛苦的两天….,需要用到TSN作者提供的pre-trained model on Kinetics, 然鹅TSN在服务器上各种编译不过, 整个实验工作都被blocked…,最终发现好用便捷的方案,记录如下

Temporal Segment Network编译记录

使用docker的好处就是不用担心安装失败等问题,可以尽情折腾

由于自己用的服务器上已经安装好了opencv, caffe,torch 等,且版本不同,因此最好的方案是用docker,但是TSN的作者没有提供docker镜像,有个tsn-docker的镜像里只装好了caffe,不可以用,因此主要使用以下三步进行编译

评论 13
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值