Skip to content

Commit 8c9e946

Browse files
committed
Update README.md
1 parent a0ad2d2 commit 8c9e946

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

README.md

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,9 +151,12 @@ mkdir data
151151
```
152152
**NOTE:** The `annotations` folder requires you to have PASCAL VOC annotations in COCO json format, which is available for download [here](https://storage.googleapis.com/coco-dataset/external/PASCAL_VOC.zip). You can also convert the XML annotatinos files to JSON by running the following script,
153153
```
154-
python tools/pascal_voc_xml2json.py
154+
python tools/pascal_voc_xml2coco_json_converter.py $VOCdevkitPATH $year
155155
```
156156
(In order to succesfully run the script above, you need to update the full path to the respective folders in the script).
157+
158+
- **Custom Dataset**
159+
Similar to above, create a directory named `CustomDataset` in the `data` folder and add symlinks to the `annotations` directory and `JPEGImages` as shown for Pascal Voc dataset.
157160

158161
Recommend to put the images on a SSD for possible better training performance
159162

@@ -231,7 +234,7 @@ Specify `—-use_tfboard` to log the losses on Tensorboard.
231234
- use `--dataset keypoints_coco2017` when training for keypoint-rcnn.
232235
- use `--dataset voc2007` when training for PASCAL VOC 2007.
233236
- use `--dataset voc2012` when training for PASCAL VOC 2012.
234-
- use `--dataset custom_dataset --num_classes $NUM_CLASSES` when training for your custom dataset. Here, `$NUM_CLASSES` is the number of object classes + 1 (for background class) present in your custom dataset.
237+
- use `--dataset custom_dataset --num_classes $NUM_CLASSES` when training for your custom dataset. Here, `$NUM_CLASSES` is the number of object classes **+ 1** (for background class) present in your custom dataset.
235238

236239
### The use of `--iter_size`
237240
As in Caffe, update network once (`optimizer.step()`) every `iter_size` iterations (forward + backward). This way to have a larger effective batch size for training. Notice that, step count is only increased after network update.

0 commit comments

Comments
 (0)