代码解析---MaskFormer

本文是在博客
https://blog.csdn.net/qq_36810544/article/details/125662056
的基础上的进一步解读注释

1 概述

整个代码结构基于detectron2框架,所以会有很多注册的指令和from_config()函数,这两个都不影响代码的逻辑,在看源码的时候不必纠结。
把所有的from_config()看成从配置文件读取相关变量的值即可,具体的值可以在config/xx/xx.yaml文件种找到。
注册指令是为了detectron2可以检测到,看源码的时候可以直接忽略这条指令。 
 

2 整体结构

所有核心代码都在MaskFormer-main/mask_former/文件夹下。
总共四个核心类分别是:

1    ./mask_former/mask_former_model.py中的MaskFormer
2    ./mask_former

你可能感兴趣的:(人工智能,计算机视觉)