generate.py 文件源码

def imdb_proposals(net, imdb, rank, count, output_dir):
    """Generate RPN proposals on all images in an imdb."""

    _t = Timer()
    for i in xrange(rank, imdb.num_images, count): # imdb.num_images
        im = cv2.imread(imdb.image_path_at(i))
        _t.tic()
        imdb_boxes, scores = im_proposals(net, im)
        with open(osp.join(output_dir, "{}.pkl".format(i)), "wb") as fp:
            cPickle.dump(imdb_boxes, fp, cPickle.HIGHEST_PROTOCOL)
        _t.toc()
        print 'im_proposals: {:d}/{:d} {:.3f}s' \
              .format(i + 1, imdb.num_images, _t.average_time)
        if 0:
            dets = np.hstack((imdb_boxes, scores))
            # from IPython import embed; embed()
            _vis_proposals(im, dets[:3, :], thresh=0.9)
            plt.show()