All Projects → shuangshuangguo → Caffe2pytorch Tsn

shuangshuangguo / Caffe2pytorch Tsn

Transform the caffe model to pytorch model for Temporal Segment Network

Programming Languages

lua
6591 projects

Projects that are alternatives of or similar to Caffe2pytorch Tsn

Unreal caffe
Self Maintained Caffe. In this version Faster-RCNN, RFCN needs layer are fully supported!
Stars: ✭ 38 (-44.93%)
Mutual labels:  caffe
Active Convolution
Active Convolution
Stars: ✭ 56 (-18.84%)
Mutual labels:  caffe
Audio Pretrained Model
A collection of Audio and Speech pre-trained models.
Stars: ✭ 61 (-11.59%)
Mutual labels:  caffe
Convolutionalemotion
A deep convolutional neural network system for live emotion detection
Stars: ✭ 40 (-42.03%)
Mutual labels:  caffe
Training toolbox caffe
Training Toolbox for Caffe
Stars: ✭ 51 (-26.09%)
Mutual labels:  caffe
Haddoc2
Caffe to VHDL
Stars: ✭ 57 (-17.39%)
Mutual labels:  caffe
Picanet
Stars: ✭ 35 (-49.28%)
Mutual labels:  caffe
Pwc Net
PWC-Net: CNNs for Optical Flow Using Pyramid, Warping, and Cost Volume, CVPR 2018 (Oral)
Stars: ✭ 1,142 (+1555.07%)
Mutual labels:  caffe
Fabrik
🏭 Collaboratively build, visualize, and design neural nets in browser
Stars: ✭ 1,081 (+1466.67%)
Mutual labels:  caffe
Head Pose Estimator.caffe
Head Pose Estimator on Caffe
Stars: ✭ 60 (-13.04%)
Mutual labels:  caffe
Darknet2caffe
Convert Darknet model to Caffe's
Stars: ✭ 46 (-33.33%)
Mutual labels:  caffe
Jacinto Ai Devkit
Training & Quantization of embedded friendly Deep Learning / Machine Learning / Computer Vision models
Stars: ✭ 49 (-28.99%)
Mutual labels:  caffe
Feathercnn
FeatherCNN is a high performance inference engine for convolutional neural networks.
Stars: ✭ 1,106 (+1502.9%)
Mutual labels:  caffe
Paperspace Python
Paperspace API for python
Stars: ✭ 38 (-44.93%)
Mutual labels:  caffe
Ssd Models
把极速检测器的门槛给我打下来make lightweight caffe-ssd great again
Stars: ✭ 62 (-10.14%)
Mutual labels:  caffe
Bottom Up Attention
Bottom-up attention model for image captioning and VQA, based on Faster R-CNN and Visual Genome
Stars: ✭ 989 (+1333.33%)
Mutual labels:  caffe
Genproto
An online GUI tool used to visualize prototxt and generate prototxt for caffe(current version).
Stars: ✭ 56 (-18.84%)
Mutual labels:  caffe
Mtcnn
face detection and alignment with mtcnn
Stars: ✭ 66 (-4.35%)
Mutual labels:  caffe
Mobilnet ssd opencv
MobilNet-SSD object detection in opencv 3.4.1
Stars: ✭ 64 (-7.25%)
Mutual labels:  caffe
Nhyai
AI智能审查,支持色情识别、暴恐识别、语言识别、敏感文字检测和视频检测等功能,以及各种OCR识别能力,如身份证、驾照、行驶证、营业执照、银行卡、手写体、车牌和名片识别等功能,可以访问网站体验功能。
Stars: ✭ 60 (-13.04%)
Mutual labels:  caffe

TSN model - Caffe2pytorch

This project transfers tsn-caffe model to tan-pytorch model.

The model dir saves my transferred model. I test them on UCF101 and HMDB51 dataset (split1), and get comparable results with the paper as follows.

Dataset RGB Flow Fusion
UCF101 86.01% 87.70% 93.82%
HMDB51 54.90% 63.53% 71.18%

This project has three steps

  • first get .hdf5file from caffemodel by export_to_hdf5.py.
  • then use the .hdf5 file to get torch model by googlenet.lua. (Because the kinetics caffe model modified the layer name, there is small change in googlenet_kinetics.lua)
  • finally transfer torch model to pytorch model by convert_torch.py.

Something to be noticed:

  • You also need to modify test_videos.py because of the problem of state_dict, please see details in test_videos.py
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].