Merge pull request #12 from ayumiymk/master

fix bug in create_svt_tfrecord.py
bgshih · Oct 22, 2018 · a7e8613 · a7e8613
2 parents 058b6bc + 8da4923
commit a7e8613
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/create_svt_tfrecord.py b/tools/create_svt_tfrecord.py
@@ -57,7 +57,7 @@ def create_svt_subset(output_path):
         fields.TfExampleFields.image_format: \
           dataset_util.bytes_feature('jpeg'.encode('utf-8')),
         fields.TfExampleFields.filename: \
-          dataset_util.bytes_feature(image_rel_path.encode('utf-8')),
+          dataset_util.bytes_feature(crop_name.encode('utf-8')),
         fields.TfExampleFields.channels: \
           dataset_util.int64_feature(3),
         fields.TfExampleFields.colorspace: \