资源分类
》 资源 》算法》detectron2-helpers

detectron2-helpers

2020-03-05 | |  70 |   0 |   0

detectron2-helpers

Helpers for detectron2

DummyAlbuMapper allows you to use albumentations during fitting detectron2 for detection tasks.

It requires improvements to work with keypoints and masks.

To use custom dataset_mapper you also need to define new class Trainer with modified build_train_loader. Smth like this:

from detectron2.engine import DefaultTrainerclass Trainer(DefaultTrainer):    @classmethod
    def build_train_loader(cls, cfg):
        mapper = DummyAlbuMapper(cfg, is_train=True)        return build_detection_train_loader(cfg, mapper=mapper)


上一篇：detectron-traffic-signs

下一篇：RetinaNet-and-SSD-in-PyTorch-Detectron

用户评价
全部评价

热门资源

  • TensorFlow-Course

    This repository aims to provide simple and read...

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • mxnet_VanillaCNN

    This is a mxnet implementation of the Vanilla C...

  • DuReader_QANet_BiDAF

    Machine Reading Comprehension on DuReader Usin...

  • Klukshu-Sockeye-...

    KLUKSHU SOCKEYE PROJECTS 2016 This repositor...