mmengine中厝村模型参数为pth的代码是哪里
时间: 2024-03-21 17:41:21 浏览: 51
PyPI 官网下载 | nomnomdata_engine-1.1.7-py3-none-any.whl
在mmdetection中,厝村模型对应的代码位于`configs/fcos/fcos_center-normbbox-centeronreg-giou_r50_caffe_fpn_gn-head_4x4_1x_coco.py`文件中。
在这个配置文件中,我们可以看到:
```python
# model settings
model = dict(
type='FCOS',
pretrained='open-mmlab://detectron2/resnet50_caffe',
backbone=dict(
type='ResNet',
depth=50,
num_stages=4,
out_indices=(0, 1, 2, 3),
frozen_stages=1,
norm_eval=True,
style='caffe'),
neck=dict(
type='FPN',
in_channels=[256, 512, 1024, 2048],
out_channels=256,
num_outs=5),
bbox_head=dict(
type='FCOSHead',
num_classes=80,
in_channels=256,
stacked_convs=4,
feat_channels=256,
strides=[8, 16, 32, 64, 128],
regress_ranges=((-1, 64), (64, 128), (128, 256), (256, 512), (512, INF)),
center_sampling=True,
center_sample_radius=1.5,
norm_on_bbox=True,
centerness_on_reg=True,
dcn_on_last_conv=False,
use_dgqp=False,
loss_cls=dict(
type='FocalLoss',
use_sigmoid=True,
gamma=2.0,
alpha=0.25,
loss_weight=1.0),
loss_bbox=dict(type='GIoULoss', loss_weight=2.0),
loss_centerness=dict(
type='CrossEntropyLoss', use_sigmoid=True, loss_weight=1.0)),
# training and testing settings
train_cfg=dict(
assigner=dict(
type='MaxIoUAssigner',
pos_iou_thr=0.5,
neg_iou_thr=0.4,
min_pos_iou=0,
ignore_iof_thr=-1),
allowed_border=-1,
pos_weight=-1,
debug=False),
test_cfg=dict(
nms_pre=1000,
min_bbox_size=0,
score_thr=0.05,
nms=dict(type='nms', iou_threshold=0.6),
max_per_img=100))
```
其中,`pretrained`参数指定了使用预训练的权重,这里设置为了`open-mmlab://detectron2/resnet50_caffe`,表示使用Detectron2模型库中的预训练权重。在进行训练时,这个权重会被加载到网络中,并作为初始化参数。在训练完成后,模型参数会被保存到.pth文件中。具体的保存代码可以在`tools/train.py`文件中找到。
阅读全文