當前位置: 首頁>>代碼示例>>Python>>正文


Python caffe.set_random_seed方法代碼示例

本文整理匯總了Python中caffe.set_random_seed方法的典型用法代碼示例。如果您正苦於以下問題:Python caffe.set_random_seed方法的具體用法?Python caffe.set_random_seed怎麽用?Python caffe.set_random_seed使用的例子?那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在caffe的用法示例。


在下文中一共展示了caffe.set_random_seed方法的2個代碼示例,這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚,您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: _init_caffe

# 需要導入模塊: import caffe [as 別名]
# 或者: from caffe import set_random_seed [as 別名]
def _init_caffe(cfg):
    """Initialize pycaffe in a training process.
    """

    import caffe
    # fix the random seeds (numpy and caffe) for reproducibility
    np.random.seed(cfg.RNG_SEED)
    caffe.set_random_seed(cfg.RNG_SEED)
    # set up caffe
    caffe.set_mode_gpu()
    caffe.set_device(cfg.GPU_ID) 
開發者ID:playerkk,項目名稱:face-py-faster-rcnn,代碼行數:13,代碼來源:train_faster_rcnn_alt_opt.py

示例2: run

# 需要導入模塊: import caffe [as 別名]
# 或者: from caffe import set_random_seed [as 別名]
def run(self):
        """This method runs in the new process."""
        global logger
        setup_exceptions()
        logger = log_utils.setup_logger('tile_worker')

        if self.caffe_path is not None:
            sys.path.append(self.caffe_path + '/python')
        if self.device >= 0:
            os.environ['CUDA_VISIBLE_DEVICES'] = str(self.device)
        import caffe
        if self.device >= 0:
            caffe.set_mode_gpu()
        else:
            caffe.set_mode_cpu()

        caffe.set_random_seed(0)
        np.random.seed(0)

        self.model = CaffeModel(*self.model_info)
        self.model.img = np.zeros((3, 1, 1), dtype=np.float32)

        while True:
            try:
                self.process_one_request()
            except KeyboardInterrupt:
                break 
開發者ID:crowsonkb,項目名稱:style_transfer,代碼行數:29,代碼來源:style_transfer.py


注:本文中的caffe.set_random_seed方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台,相關代碼片段篩選自各路編程大神貢獻的開源項目,源碼版權歸原作者所有,傳播和使用請參考對應項目的License;未經允許,請勿轉載。