thtang/Listen_Translate

ML2017 fall Final

Listen and Translate

Team: NTU_r06946003_鬼氏企業 / r06946003 湯忠憲 r06922030 傅敏桓 r03902085 吳家謙 d05921027 張鈞閔

Package :

keras==2.0.8   numpy==1.13.3   pandas==0.20.3   sklearn==0.19.1  

Usage :

Training:

cd src
python3 inference.py [train_data_path] [test_data_path] [train_caption_path] [test_caption_path] [output_path] [GPU number]

Testing:

cd src
python3 inference.py [train_data_path] [test_data_path] [train_caption_path] [test_caption_path] [output_path]

Note :

The model was stored in ./model folder.
In my .py script, I used the following script to assign the task running on GPU 0.

import os
os.environ['CUDA_VISIBLE_DEVICES'] = "0"

model architecture :

RNN retrieval model RNN+CNN retrieval model