mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
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
|