fix vc bug (#1297)

fluid/video_classification/README.md

@@ -111,7 +111,6 @@ According to the congfiguration of evaluation, the output log is like:
 Inference is used to get prediction score or video features based on trained models.
 ```
 python infer.py \
-    --batch_size=128 \
     --class_dim=101 \
     --image_shape=3,224,224 \
     --with_mem_opt=True \

fluid/video_classification/data/generate_train_data.py

@@ -9,14 +9,16 @@ for line in f.readlines():
     dd[name.lower()] = int(label) - 1
 f.close()
 
-# generate pkl
-path = 'train/'
-savepath = 'train_pkl/'
-if not os.path.exists(savepath):
+
+def generate_pkl(mode):
+    # generate pkl
+    path = '%s/' % mode
+    savepath = '%s_pkl/' % mode
+    if not os.path.exists(savepath):
         os.makedirs(savepath)
 
-fw = open('train.list', 'w')
-for folder in os.listdir(path):
+    fw = open('%s.list' % mode, 'w')
+    for folder in os.listdir(path):
         vidid = folder.split('_', 1)[1]
         this_label = dd[folder.split('_')[1].lower()]
         this_feat = []
@@ -31,5 +33,9 @@ for folder in os.listdir(path):
         cPickle.dump(res, outp, protocol=cPickle.HIGHEST_PROTOCOL)
         outp.close()
 
-    fw.write('data/train_pkl/%s.pkl\n' % vidid)
-fw.close()
+        fw.write('data/%s/%s.pkl\n' % (savepath, vidid))
+    fw.close()
+
+
+generate_pkl('train')
+generate_pkl('test')

fluid/video_classification/reader.py

@@ -16,8 +16,8 @@ THREAD = 8
 BUF_SIZE = 1024
 
 TRAIN_LIST = 'data/train.list'
-TEST_LIST = 'data/val.list'
-INFER_LIST = 'data/val.list'
+TEST_LIST = 'data/test.list'
+INFER_LIST = 'data/test.list'
 
 img_mean = np.array([0.485, 0.456, 0.406]).reshape((3, 1, 1))
 img_std = np.array([0.229, 0.224, 0.225]).reshape((3, 1, 1))

fluid/video_classification/train.py

@@ -2,6 +2,7 @@ import os
 import numpy as np
 import time
 import sys
+import paddle.v2 as paddle
 import paddle.fluid as fluid
 from resnet import TSN_ResNet
 import reader