prakrutpatel commited on
Commit
08f1bfe
·
1 Parent(s): b01c399
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -61,6 +61,7 @@ test_record_fname = dataset + '/test/tortoise.tfrecord'
61
  train_record_fname = dataset + '/train/tortoise.tfrecord'
62
  label_map_pbtxt_fname = dataset + '/train/tortoise_label_map.pbtxt'
63
  PATH_TO_LABELS = dataset + '/train/tortoise_label_map.pbtxt'
 
64
  category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=False)
65
 
66
 
 
61
  train_record_fname = dataset + '/train/tortoise.tfrecord'
62
  label_map_pbtxt_fname = dataset + '/train/tortoise_label_map.pbtxt'
63
  PATH_TO_LABELS = dataset + '/train/tortoise_label_map.pbtxt'
64
+ print(os.listdir(dataset + '/train'))
65
  category_index = label_map_util.create_category_index_from_labelmap(PATH_TO_LABELS, use_display_name=False)
66
 
67