从Faster RCNN开始学python（三）

最新推荐文章于 2023-12-02 09:15:00 发布

白色字体

最新推荐文章于 2023-12-02 09:15:00 发布

阅读量659

点赞数 1

分类专栏：从Faster RCNN开始学python

本文链接：https://blog.csdn.net/qq_36549512/article/details/80762526

版权

今天准备解析的代码如下：

import multiprocessing as mp
# queue for communicated results between processes
mp_queue = mp.Queue()
# solves, iters, etc. for each training stage
solvers, max_iters, rpn_test_prototxt = get_solvers(args.net_name)

print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'
print 'Stage 1 RPN, init from ImageNet model'
print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~'

cfg.TRAIN.SNAPSHOT_INFIX = 'stage1'
mp_kwargs = dict(
        queue=mp_queue,
        imdb_name=args.imdb_name,
        init_model=args.pretrained_model,
        solver=solvers[0],
        max_iters=max_iters[0],
        cfg=cfg)
p = mp.Process(target=train_rpn, kwargs=mp_kwargs)
p.start()
rpn_stage1_out = mp_queue.get()
p.join()

在Faster RCNN中使用的是多进程的方式，multiprocessing是使用多进程的一个包。
在multiprocessing中Queue是mp库当中用来提供多进程对象交换的方式，它将对象从一个进程传输