YOLT论文代码映射

今天下午看了看论文,大致明白了网络结构,具体还要看代码。开始分析代码:
1.首先分析yolt2.py文件
此文件中除了main函数,总共有16个子函数,应该是调来调去。先看main函数,依次调用6个函数:
(1)init_args():专门用于接收和设置参数,往往需要被设置的参数会根据接收的参数来设置。这些参数一般都是路径或者配置参数等等。
如:项目路径为 --yolt_dir
一般会根据接收的项目路径设置其他路径。
总而言之,言而总之,这个函数就是专门用于设置各种而存在的。
(2)make_label_images():
实参:
args.label_image_dir: args.data_dir + ‘category_label_images/’
args.object_labels: [boat,car]
形参:
root_dir: 此时为args.data_dir + ‘category_label_images/’
new_labels:此时为[boat,car]
这个函数又调用了函数run_cmd(cmd),也不知道是干嘛用的。
(3)recompile_darknet()
此时形参实际值为args.yolt_dir
这个函数也调用了函数run_cmd(cmd),估计是执行命令的意思吧
(4)replace_yolt_vals()
此时形参实际值为args,传入args,就相当于init_args()中所有参数了。
这个参数实在太多了,不太好看,先待定。
(5)yolt_command():此函数根据输入参数简单定义了一些变量,最后这些变量为c_arg_list这个列表服务。
(6)run_valid():不用说,此函数肯定是重要的函数。
1)调用函数split_valid_im():
参数跳来跳去,最后可以归结于init_args()函数:
参数 默认值 说明
–mode ‘test’ help="[compile, test, train, valid]"
–gpu 0 help=“GPU number, set < 0 to turn off GPU support”
–nbands 3 help=“Number of input bands (e.g.: for RGB use 3)”
–outname tmp help=“unique name of output”
–cfg_file yolo.cfg

猜你喜欢

转载自blog.csdn.net/qq_42278791/article/details/90979556
今日推荐