Skip to content

Commit 88aea9c

Browse files
author
Chuck Cho
committed
Adding mean cube file in numpy format
1 parent 08e93b5 commit 88aea9c

File tree

3 files changed

+5
-6
lines changed

3 files changed

+5
-6
lines changed

configure.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
import os
22

33
# Directories
4-
DIR_HOME = '/media/6TB'
4+
DIR_HOME = '/media/6TB/Videos'
55
DIR_CKPT = 'checkpoints'
66
DIR_LST = 'lists'
77
DIR_MODEL = 'models'
@@ -13,8 +13,8 @@
1313
DIR_DATA = os.path.join(DIR_HOME, 'UCF-101')
1414

1515
# Lists
16-
PTH_TRAIN_LST = os.path.join(DIR_LST, 'train_01.lst')
17-
PTH_EVAL_LST = os.path.join(DIR_LST, 'test_01.lst')
16+
PTH_TRAIN_LST = '/media/6TB/Videos/UCF-101/train_test_split/c3d_ucf101_train_split1.txt'
17+
PTH_EVAL_LST = '/media/6TB/Videos/UCF-101/train_test_split/c3d_ucf101_test_split1.txt'
1818
PTH_CATEGORIES = os.path.join(DIR_LST, 'categories.lst')
1919
#IMAGE_FORMAT = '{:06d}.jpg'
2020
IMAGE_FORMAT = 'image_{:04d}.jpg'

models/train01_16_128_171_mean.npy

8.02 MB
Binary file not shown.

trainer_c3d.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,7 @@ def do_eval(
132132
tmp_ = sess.run(eval_correct, feed_dict=fd)
133133
# If data was partially filled in a batch, count only those
134134
num_valid = stop_idx - start_idx
135-
tmp_ = tmp_[0:num_valid]
136-
#print "[Debug] eval_correct={}".format(tmp_)
135+
tmp_ = tmp_[:num_valid]
137136
true_count_per_batch = sess.run(tf.reduce_sum(tmp_))
138137
true_count += true_count_per_batch
139138
start_idx = stop_idx
@@ -156,7 +155,7 @@ def run_training(
156155
tag):
157156

158157
# For training, subsample training/eval data sets (if >1)
159-
subsample_rate = 1
158+
subsample_rate = 10
160159
# For periodic evaluation, subsample training/eval data sets (if >1)
161160
eval_subsample_rate = 100
162161

0 commit comments

Comments
 (0)