Commit a778063

committed

updated create_coco_tf_record.py file to include masks by default

1 parent 0593d42 commit a778063Copy full SHA for a778063

File tree

-3

lines changed

-3

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -147,7 +147,7 @@ python labelme2coco.py test test.json`
`147`	`147`	`Now we can create the TFRecord files using the [create_coco_tf_record.py script](create_coco_tf_record.py).`
`148`	`148`
`149`	`149`	```bash
`150`		`-python create_coco_tf_record.py --logtostderr --train_image_dir=images/train --test_image_dir=images/test --train_annotations_file=images/train.json --test_annotations_file=images/test.json --output_dir=./`
	`150`	`+python create_coco_tf_record.py --logtostderr --train_image_dir=images/train --test_image_dir=images/test --train_annotations_file=images/train.json --test_annotations_file=images/test.json --include_masks=True --output_dir=./`
`151`	`151`	```
`152`	`152`
`153`	`153`	`After executing this command, you should have a train.record and test.record file inside your object detection folder.`

Lines changed: 2 additions & 2 deletions

Original file line number	Diff line number	Diff line change
`@@ -51,9 +51,9 @@`
`51`	`51`
`52`	`52`
`53`	`53`	`flags = tf.app.flags`
`54`		`-tf.flags.DEFINE_boolean('include_masks', False,`
	`54`	`+tf.flags.DEFINE_boolean('include_masks', True,`
`55`	`55`	`'Whether to include instance segmentations masks '`
`56`		`- '(PNG encoded) in the result. default: False.')`
	`56`	`+ '(PNG encoded) in the result. default: True.')`
`57`	`57`	`tf.flags.DEFINE_string('train_image_dir', '',`
`58`	`58`	`'Training image directory.')`
`59`	`59`	`tf.flags.DEFINE_string('test_image_dir', '',`

Comments

(0)