import owl import sys owl.initialize(sys.argv) owl.create_cpu_device() gpu0 = owl.create_gpu_device(0) gpu1 = owl.create_gpu_device(1) owl.set_device(gpu0)
import sys, argparse
import time
import numpy as np
import owl
from owl.net.trainer import FeatureExtractor
if __name__ == "__main__":
# parse command line arguments
parser = argparse.ArgumentParser()
parser.add_argument('solver_file', help='caffe solver configure file')
parser.add_argument('snapshot', help='the snapshot idx to start from', action='store', type=int, default=0)
parser.add_argument('layer_name', help='layer_name')
parser.add_argument('feature_path', help='feature_path')
parser.add_argument('-g', '--gpu_idx', help='gpu to use', action='store', type=int, default=0)
(args, remain) = parser.parse_known_args()
solver_file = args.solver_file
layer_name = args.layer_name
feature_path = args.feature_path
gpu_idx = args.gpu_idx
snapshot = args.snapshot
print ' === Extract feature of snapshot %d, using gpu #%d === ' % (snapshot, gpu_idx)
sys_args = [sys.argv[0]] + remain
owl.initialize(sys_args)
extractor = FeatureExtractor(solver_file, snapshot, gpu_idx)
extractor.build_net()
extractor.run(layer_name, feature_path)
