DETR

源码获取：https://gitee.com/fgy120/DETR 首先对DETR做个简单介绍 目录 main函数 (一) 参数设置 main函数（二）搭建模型 build_backbone()：包括构建位置编码器以及backbone 上图即为DETR的流程pipeline，相比以前的RCNN系列、YOLO系列等，最特别的在于加入了Transformer。 直接看源码，从train.py的主函数开始。 if __name__ == '__main__': parser = argparse.ArgumentParser('DETR training and evaluation script', parents=[get_args_parser()]) args = parser.parse_args() if args.output_dir: Path(args.output_dir).mkdir(parents=True, exist_ok=True)#以output_dir创建Path对象并执行mkdir创建文件夹操作 main(args) 首先是常规的参数解析操作，利用的argparse库，主要通过解析命令行输入的参数来设置模型训练的超参数或其他设置。第一步创建解析对象parser，运行parser.parse_args方法得到解析后的各个参数args，默认为解析运行代码的命令行。如果其中包含output_dir参数且output_dir不存在，利用Pathlib中的Path库的mkdir方法创建output_dir的路径文件夹。 Path(args.output_dir).mkdir(parents=True, exist_ok=True) parents：如果父目录不存在，是否创建父目录。 exist_ok：只有在目录不存在时创建目录，目录已存在时不会抛出异常。 argparse具体介绍可以看这篇。 argparse解析器_在努力的松鼠的博客-CSDN博客 argparse是一个Python模块，用来解析命令行参数，主要有三个步骤：1、创建 ArgumentParser() 对象2、调用 add_argument() 方法添加参数3、使用 parse_args() 解析添加的参数一、创建解析器代码示例：parser = argparse.ArgumentParser('DETR training and evaluation script',