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)
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. #
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)