InsegTrainer#
- class InsegTrainer(cfg)[source]#
Bases:
TrainerTrainer for instance segmentation with instance-aware collation.
Identical to
Trainerexcept the train and val loaders useinseg_collate_fn(which preserves variable per-sample instance/query targets and appliesmix_probonly at train time) over a stateful, resume-able sampler/loader. Use with the instance-segmentation losses (e.g.FastInstanceSegmentationLoss). Registered asInsegTrainer– select viatrain = dict(type="InsegTrainer").