preparing_inputs.md 1.66 KB
Newer Older
1
2
3
4
# Preparing Inputs

Tensorflow Object Detection API reads data using the TFRecord file format. Two
sample scripts (`create_pascal_tf_record.py` and `create_pet_tf_record.py`) are
5
provided to convert from the PASCAL VOC dataset and Oxford-IIIT Pet dataset to
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
TFRecords.

## Generating the PASCAL VOC TFRecord files.

The raw 2012 PASCAL VOC data set can be downloaded
[here](http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCtrainval_11-May-2012.tar).
Extract the tar file and run the `create_pascal_tf_record` script:

```
# From tensorflow/models/object_detection
tar -xvf VOCtrainval_11-May-2012.tar
./create_pascal_tf_record --data_dir=VOCdevkit \
    --year=VOC2012 --set=train --output_path=pascal_train.record
./create_pascal_tf_record --data_dir=/home/user/VOCdevkit \
    --year=VOC2012 --set=val --output_path=pascal_val.record
```

You should end up with two TFRecord files named pascal_train.record and
pascal_val.record in the tensorflow/models/object_detection directory.

The label map for the PASCAL VOC data set can be found at
data/pascal_label_map.pbtxt.

29
## Generation the Oxford-IIIT Pet TFRecord files.
30

31
The Oxford-IIIT Pet data set can be downloaded from
32
33
34
35
36
37
38
39
40
41
42
43
44
45
[their website](http://www.robots.ox.ac.uk/~vgg/data/pets/). Extract the tar
file and run the `create_pet_tf_record` script to generate TFRecords.

```
# From tensorflow/models/object_detection
tar -xvf annotations.tar.gz
tar -xvf images.tar.gz
./create_pet_tf_record --data_dir=`pwd` --output_dir=`pwd`
```

You should end up with two TFRecord files named pet_train.record and
pet_val.record in the tensorflow/models/object_detection directory.

The label map for the Pet dataset can be found at data/pet_label_map.pbtxt.