mirror of https://github.com/open-mmlab/mmcv.git
17 lines
584 B
Markdown
17 lines
584 B
Markdown
|
## CNN
|
||
|
|
|
||
|
|
The subpackage `mmcv.cnn` ships with lots of CNN backbones and the number is growing.
|
||
|
|
Different from models provided in `torchvision`, these models are designed
|
||
|
|
for actual usage in different cases, with highly customizable interfaces.
|
||
|
|
|
||
|
|
Taking ResNet as an example, you can specify the following things:
|
||
|
|
|
||
|
|
- number of stages
|
||
|
|
- stage of output feature maps
|
||
|
|
- location of the stride 2 convolutional layer
|
||
|
|
- strides and dilation of each stage
|
||
|
|
- whether to freeze some first stages
|
||
|
|
- whether to fix BN stats
|
||
|
|
- whether to fix BN parameters
|
||
|
|
- whether to use checkpoint to save memory
|