Skip to content

Latest commit

 

History

History
50 lines (37 loc) · 1.13 KB

README.md

File metadata and controls

50 lines (37 loc) · 1.13 KB

Shot_detection

CNN+LSTMにてショットの自動分割を行う

Code

for Setup

学習を行うまでの準備に使うコード

conformity.py

ラベリングの修正

marge.py

各動画から抽出したCSVファイルを一つに統合する

all_extra_featue.py / extra_feature.py

学習済みCNNから特徴ベクトルを抽出するPG モデルと動画を設定する必要がある

extractor.py

学習済みCNNをロードしてくるクラス ここに新しいCNNモデルなんかを今後追加して書いてくといいかもね

for DeepLearning

学習で使用するコード

LSTM.py

LSTMのモデルと学習を行う

3Dconv.py

3D Convolutional Neural Networks 3次元畳込みニューラルネットワークでのショット分類

load.py

CSVのファイルを読み込むクラス

for Attention Visualize

Attentionによる可視化

attention_visualize.py

可視化を行うスクリプト

load_forAttentionVis.py

Requirement

  • Software
    • python3.6.3
    • tensorflow==1.7.0
    • keras==2.1.5
    • numpy==1.14.0
    • matplotlib==2.2.2
    • opencv-python==3.4.1.15