20 lines
1.4 KiB
Markdown
20 lines
1.4 KiB
Markdown
# Develop
|
|
|
|
This document provides tutorials to develop CenterNet. `lib/src/opts` lists a few more options that the current version supports.
|
|
|
|
## New dataset
|
|
Basically there are three steps:
|
|
|
|
- Convert the dataset annotation to [COCO format](http://cocodataset.org/#format-data). Please refer to [src/tools/convert_kitti_to_coco.py](../src/tools/convert_kitti_to_coco.py) for an example to convert kitti format to coco format.
|
|
- Create a dataset intilization file in `src/lib/datasets/dataset`. In most cases you can just copy `src/lib/datasets/dataset/coco.py` to your dataset name and change the category information, and annotation path.
|
|
- Import your dataset at `src/lib/datasets/dataset_factory`.
|
|
|
|
## New task
|
|
|
|
You will need to add files to `src/lib/datasets/sample/`, `src/lib/datasets/trains/`, and `src/lib/datasets/detectors/`, which specify the data generation during training, the training targets, and the testing, respectively.
|
|
|
|
## New architecture
|
|
|
|
- Add your model file to `src/lib/models/networks/`. The model should accept a dict `heads` of `{name: channels}`, which specify the name of each network output and its number of channels. Make sure your model returns a list (for multiple stages. Single stage model should return a list containing a single element.). The element of the list is a dict contraining the same keys with `heads`.
|
|
- Add your model in `model_factory` of `src/lib/models/model.py`.
|