总结一下最近一年来阅读目标检测框架的一些经验,主要是基于facebook的maskrcnn_benchmark和Detectron2。
组件
目标检测的框架需要包含:
组件 | 功能 |
---|---|
config | 所有参数设置 |
structures | 数据容器,image, box, label, mask, keypoint |
data | 数据整理,dataset, dataloader, transform |
modeling | 网络模型,meta_arch, backbone, proposals_generator, anchor_generator, heads |
layers | 特殊模块,roi_align, batch_norm, defrom_conv, separable_conv |
engien | 训练推理, trainer, inference |
solver | 优化器, optimizer, lr_scheduler |
evaluation | 评估器, 各种数据集的evaluator |
utils | 其他一些实现,model_zoo, checkpointer, logger, rigistry, 分布式方法等 |
从宏观上看,阅读分析目标检测框架时可以按照以上顺序逐步进行,同时结合训练和推理的demo,在pycharm下逐步调试。