Python Faster-RCNN Data Interface Generator
The purpose of this tool is to generate dataset interface of Faster-RCNN(FRCN) that can help you to deploy training on your own dataset as fast as possible.
This tool will generate dataset interface like
Also a set of net prototxt files like
And a bash script similar to
Before using this generator, please make sure you have already set up a voc-like tree structure of your dataset(in your
PASCAL_VOC_devkit └── data ├── Annotations │ └── *.xml │ ├── ImageSets │ └── Main │ ├── train.txt │ └── test.txt └── JPEGImages └── *.jpg
Then you can run the
generate.py to generate interface, prototxt and script files
--froot: absolute path to your faster-rcnn root.
--idname: name your dataset, whatever you want. It will be used in the future training.
--cls: class names in your dataset, use comma to split them (e.g: cat,dog,tiger)
--dvkt: absolute path to your dataset devkit(mentioned above)
Please notice that, the tool will do everything automatically for you except for generating of
In case of changing this
factory.py file without any backup, the tool will not overwrite it but create a new
<idname>_factory.py file for you.
You have 2 options to complete the last step manually:
- Backup the original
factory.pyfile and replace it with the generating one.
- Copy the code block displayed in the task summary into the