OrgLion-ML
/
CenterNet
mirror of https://github.com/xingyizhou/CenterNet.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Update DEVELOP.md

This commit is contained in:
Guillaume Caya-Letourneau 2019-10-06 10:58:02 -04:00 committed by GitHub
parent 9efed05d53
commit 5e17c70bc7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
readme/DEVELOP.md
View File

@ -15,5 +15,5 @@ You will need to add files to `src/lib/datasets/sample/`, `src/lib/datasets/trai
## 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 channals. 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 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`.