目标检测框架阅读步骤

Posted 2021-03-10 zi-wang

总结一下最近一年来阅读目标检测框架的一些经验，主要是基于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下逐步调试。