Skip to content

怎么只训练目标检测这一个任务 #220

Description

@yangoono

我的配置文件default.py是这样的

if training 3 tasks end-to-end, set all parameters as True

Alternating optimization

_C.TRAIN.SEG_ONLY = False # Only train two segmentation branchs
_C.TRAIN.DET_ONLY =True # Only train detection branch
_C.TRAIN.ENC_SEG_ONLY = False # Only train encoder and two segmentation branchs
_C.TRAIN.ENC_DET_ONLY = False # Only train encoder and detection branch

Single task

_C.TRAIN.DRIVABLE_ONLY = False # Only train da_segmentation task
_C.TRAIN.LANE_ONLY = False # Only train ll_segmentation task
_C.TRAIN.DET_ONLY = True # Only train detection task

但是训练的过程中却出现了Driving area Segment: Acc(0.366) IOU (0.181) mIOU(0.222)
Lane line Segment: Acc(0.641) IOU (0.019) mIOU(0.147)
Detect: P(0.022) R(0.699) mAP@0.5(0.143) mAP@0.5:0.95(0.033)
看样子三个任务都在进行。是不是我还有什么地方忘记改了

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions