Python set_phase_trainの例

コード例 #1

ファイル: rerun.py プロジェクト: visme07/dl-playground

import sys
import csv, os
import numpy as np
import time
sys.path.append('~/caffe/caffe/python/caffe')
caffe_root = '/home/dueo/caffe/caffe/'
sys.path.insert(0, caffe_root + 'python')
import caffe

predLayerName = 'fc8'  #Output of the last inner product layer
description = 'train_val.prototxt'
learnedModel = 'models/alexnet_128_test_iter_2900.caffemodel'

caffe.set_mode_gpu()
caffe.set_phase_train()  #TODO delete
net = caffe.Net(description, learnedModel)

solver = caffe.SGDSolver('solver.auto.prototxt')
solver.net.copy_from(net)

コード例 #2

    if args.gpu >= 0:
        isModeCPU = False   # command line overrides solver file
        gpuId = args.gpu
    else:
        isModeCPU = (solverParam.solver_mode == solverParam.SolverMode.Value('CPU'))
        gpuId = 0
        
    # Note that different Caffe APIs put this function different places (module vs net object)
    print('[deploy]: CPU mode = %s' % isModeCPU)
    try:
        if not isModeCPU:
            caffe.set_mode_gpu()
            caffe.set_device(gpuId)
        else:
            caffe.set_mode_cpu()
        caffe.set_phase_train()
    except AttributeError:
        if not isModeCPU:
            net.set_mode_gpu()
            net.set_device(gpuId)
        else:
            net.set_mode_cpu()
        net.set_phase_train()

    #----------------------------------------
    # Do it
    #----------------------------------------
    Yhat = _eval_cube(net, X, Mask, batchDim, bandSpec=[args.lb, args.ub])
 
    # Apply thresholds and chuck the border before saving.
    #

コード例 #3

ファイル: rerun.py プロジェクト: zjZSTU/dl-playground

import sys
import csv, os
import numpy as np
import time

sys.path.append("~/caffe/caffe/python/caffe")
caffe_root = "/home/dueo/caffe/caffe/"
sys.path.insert(0, caffe_root + "python")
import caffe

predLayerName = "fc8"  # Output of the last inner product layer
description = "train_val.prototxt"
learnedModel = "models/alexnet_128_test_iter_2900.caffemodel"

caffe.set_mode_gpu()
caffe.set_phase_train()  # TODO delete
net = caffe.Net(description, learnedModel)


solver = caffe.SGDSolver("solver.auto.prototxt")
solver.net.copy_from(net)