import sys
from optparse import OptionParser
sys.path.append('./')
import yolo
from yolo.utils.process_config import process_config
# parser初始化
parser = OptionParser()
# 为命令行添加选项
parser.add_option("-c", "--conf", dest="configure",
help="configure filename")
(options, args) = parser.parse_args()
if options.configure:
conf_file = str(options.configure)
else:
print('please sspecify --conf configure filename')
exit(0)
# 读取config参数,返回4个字典
common_params, dataset_params, net_params, solver_params = process_config(conf_file)
dataset = eval(dataset_params['name'])(common_params, dataset_params)
net = eval(net_params['name'])(common_params, net_params)
solver = eval(solver_params['name'])(dataset, net, common_params, solver_params)
solver.solve()
nillboy/yolo代码解读3:/tools/train.py
猜你喜欢
转载自blog.csdn.net/weixin_38900691/article/details/79586516
今日推荐
周排行