base = '../faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py'
model = dict(
roi_head=dict(
type='DynamicRoIHead',
bbox_head=dict(
type='Shared2FCBBoxHead',
in_channels=256,
fc_out_channels=1024,
roi_feat_size=7,
num_classes=1,
bbox_coder=dict(
type='DeltaXYWHBBoxCoder',
target_means=[0., 0., 0., 0.],
target_stds=[0.1, 0.1, 0.2, 0.2]),
reg_class_agnostic=False,
loss_cls=dict(
type='CrossEntropyLoss', use_sigmoid=False, loss_weight=1.0),
loss_bbox=dict(type='SmoothL1Loss', beta=1.0, loss_weight=1.0))))
train_cfg = dict(
rpn_proposal=dict(nms_thr=0.85),
rcnn=dict(
dynamic_rcnn=dict(
iou_topk=75,
beta_topk=10,
update_iter_interval=100,
initial_iou=0.4,
initial_beta=1.0)))
test_cfg = dict(rpn=dict(nms_thr=0.85))
Traceback (most recent call last):
File "train.py", line 153, in
main()
File "train.py", line 149, in main
meta=meta)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/apis/train.py", line 128, in train_detector
runner.run(data_loaders, cfg.workflow, cfg.total_epochs)
File "/home/esec/.local/lib/python3.6/site-packages/mmcv/runner/epoch_based_runner.py", line 122, in run
epoch_runner(data_loaders[i], **kwargs)
File "/home/esec/.local/lib/python3.6/site-packages/mmcv/runner/epoch_based_runner.py", line 32, in train
**kwargs)
File "/home/esec/.local/lib/python3.6/site-packages/mmcv/parallel/data_parallel.py", line 31, in train_step
return self.module.train_step(*inputs[0], **kwargs[0])
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/detectors/base.py", line 236, in train_step
losses = self(**data)
File "/home/esec/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/core/fp16/decorators.py", line 51, in new_func
return old_func(*args, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/detectors/base.py", line 171, in forward
return self.forward_train(img, img_metas, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/detectors/two_stage.py", line 164, in forward_train
**kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/roi_heads/dynamic_roi_head.py", line 89, in forward_train
img_metas)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/roi_heads/dynamic_roi_head.py", line 128, in _bbox_forward_train
*bbox_targets)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/core/fp16/decorators.py", line 131, in new_func
return old_func(*args, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/roi_heads/bbox_heads/bbox_head.py", line 182, in loss
reduction_override=reduction_override)
File "/home/esec/.local/lib/python3.6/site-packages/torch/nn/modules/module.py", line 541, in call
result = self.forward(*input, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/losses/smooth_l1_loss.py", line 93, in forward
**kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/losses/utils.py", line 94, in wrapper
loss = loss_func(pred, target, **kwargs)
File "/home/esec/hewei/kaggle/mmdetection/mmdet/models/losses/smooth_l1_loss.py", line 21, in smooth_l1_loss
assert beta > 0
AssertionError