update

Author: Abdul Dakkak

vision/tools/create_pet_tf_record.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 # ==============================================================================
 
+# python create_pet_tf_record.py --label_map_path=../support/object_detection/assets/pet_label_map.pbtxt --data_dir=$HOME/data/pet --output_dir=$HOME/data/pet
+
 r"""Convert the Oxford pet dataset to TFRecord for object_detection.
 
 See: O. M. Parkhi, A. Vedaldi, A. Zisserman, C. V. Jawahar
@@ -54,7 +56,7 @@
                      'in the latter case, the resulting files are much larger.')
 flags.DEFINE_string('mask_type', 'png', 'How to represent instance '
                     'segmentation masks. Options are "png" or "numerical".')
-flags.DEFINE_integer('num_shards', 10, 'Number of TFRecord shards')
+flags.DEFINE_integer('num_shards', 1, 'Number of TFRecord shards')
 
 FLAGS = flags.FLAGS