cnn.md 584 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
## 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