Skip to content
Snippets Groups Projects
Unverified Commit 0d74cd8f authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Merge pull request #21 from hellock/docs

Add some instructions of customizing models
parents 6efefa27 6f428389
No related branches found
No related tags found
No related merge requests found
...@@ -30,6 +30,50 @@ such as `SingleStageDetector` and `TwoStageDetector`. ...@@ -30,6 +30,50 @@ such as `SingleStageDetector` and `TwoStageDetector`.
Following some basic pipelines (e.g., two-stage detectors), the model structure Following some basic pipelines (e.g., two-stage detectors), the model structure
can be customized through config files with no pains. can be customized through config files with no pains.
If we want to implement some new components, e.g, the path aggregation
FPN structure in [Path Aggregation Network for Instance Segmentation](https://arxiv.org/abs/1803.01534), there are two things to do.
1. create a new file in `mmdet/models/necks/pafpn.py`.
```python
class PAFPN(nn.Module):
def __init__(self,
in_channels,
out_channels,
num_outs,
start_level=0,
end_level=-1,
add_extra_convs=False):
pass
def forward(self, inputs):
# implementation is ignored
pass
```
2. modify the config file from
```python
neck=dict(
type='FPN',
in_channels=[256, 512, 1024, 2048],
out_channels=256,
num_outs=5)
```
to
```python
neck=dict(
type='PAFPN',
in_channels=[256, 512, 1024, 2048],
out_channels=256,
num_outs=5)
```
We will release more components (backbones, necks, heads) for research purpose.
### Write a new model ### Write a new model
To write a new detection pipeline, you need to inherit from `BaseDetector`, To write a new detection pipeline, you need to inherit from `BaseDetector`,
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment