资源算法C3D-TF

C3D-TF

2020-02-10 | |  36 |   0 |   0

C3D-TF

C3D-tensorflow version, used for my gesture training.

model

There is a c3d-model transformed from c3d-facebook project model file.(5 conv 2 fc, little like vgg). Here is the file. c3d.model Download and store it in the root folder. resnet50_weights_tf_dim_ordering_tf_kernels.h5 Download and store it in the root folder.

file directory

  • ./ -> root

    • C3DModel.py -> C3D model

    • CNNLSTM.py -> CNN+LSTM interface

    • LoadPCKModel.py -> load params for C3D

    • mobilenet.py -> my mobilenet define

    • mobilenet_v1.py -> official slim definition

    • resnet50.py -> my resnet50 define

    • utils.py -> some common interface for tf

    • opticalflow_gpu/ gpu version

    • GetTrain.py -> data balance. input:label.txt output:gen_label_train.txt gen_label_test.txt

    • OpticalFlow.py -> opticalflow opencv

    • data_augment.py -> offline data augmentation

    • extract_video_frame_2_dirs.py -> origin dataset process

    • generate_label.py

    • shuffle_label.py -> label shuffle

    • DataSet -> dataProcessScripts

    • Net -> model definition(C3D,ResNet,MobileNet,LSTM)

    • demo.py -> demo for real-time video process

    • input_data.py -> data process when training or testing

    • preprocess.py -> img distortion function, which is not used now.

    • test.py -> VIVA Gesture dataset test

    • train_c3d_ucf101.py -> c3d training process

    • train_lstm.py/train_resnet_lstm.py -> train resnet+lstm training process

    • train_mobilenet_lstm.py -> mobilenet+lstm training process

    How to use my scripts?

    1. if you use VIVA Gesture Dataset, use scripts under DataSet to perform offline dataset process, otherwise, you should write codes yourself to process Dataset, and use function in extract_video_frame_2_dirs.py

    2. Then use train_*.py to train our models. If you want to change the model, please refer CNNLSTM.py and resnet50.py


    上一篇:New-C3D-Caffe

    下一篇:nn-c3d

    用户评价
    全部评价

    热门资源

    • seetafaceJNI

      项目介绍 基于中科院seetaface2进行封装的JAVA...

    • spark-corenlp

      This package wraps Stanford CoreNLP annotators ...

    • Keras-ResNeXt

      Keras ResNeXt Implementation of ResNeXt models...

    • capsnet-with-caps...

      CapsNet with capsule-wise convolution Project ...

    • inferno-boilerplate

      This is a very basic boilerplate example for pe...