File tree Expand file tree Collapse file tree 3 files changed +5
-6
lines changed Expand file tree Collapse file tree 3 files changed +5
-6
lines changed Original file line number Diff line number Diff line change 1
1
import os
2
2
3
3
# Directories
4
- DIR_HOME = '/media/6TB'
4
+ DIR_HOME = '/media/6TB/Videos '
5
5
DIR_CKPT = 'checkpoints'
6
6
DIR_LST = 'lists'
7
7
DIR_MODEL = 'models'
13
13
DIR_DATA = os .path .join (DIR_HOME , 'UCF-101' )
14
14
15
15
# 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'
18
18
PTH_CATEGORIES = os .path .join (DIR_LST , 'categories.lst' )
19
19
#IMAGE_FORMAT = '{:06d}.jpg'
20
20
IMAGE_FORMAT = 'image_{:04d}.jpg'
Original file line number Diff line number Diff line change @@ -132,8 +132,7 @@ def do_eval(
132
132
tmp_ = sess .run (eval_correct , feed_dict = fd )
133
133
# If data was partially filled in a batch, count only those
134
134
num_valid = stop_idx - start_idx
135
- tmp_ = tmp_ [0 :num_valid ]
136
- #print "[Debug] eval_correct={}".format(tmp_)
135
+ tmp_ = tmp_ [:num_valid ]
137
136
true_count_per_batch = sess .run (tf .reduce_sum (tmp_ ))
138
137
true_count += true_count_per_batch
139
138
start_idx = stop_idx
@@ -156,7 +155,7 @@ def run_training(
156
155
tag ):
157
156
158
157
# For training, subsample training/eval data sets (if >1)
159
- subsample_rate = 1
158
+ subsample_rate = 10
160
159
# For periodic evaluation, subsample training/eval data sets (if >1)
161
160
eval_subsample_rate = 100
162
161
You can’t perform that action at this time.
0 commit comments