channel_swap=channel_swap)
if args.gpu:
- print('GPU mode')
+ print("GPU mode")
+ else:
+ print("CPU mode")
# Load numpy array (.npy), directory glob (*.jpg), or image file.
args.input_file = os.path.expanduser(args.input_file)
if args.input_file.endswith('npy'):
+ print("Loading file: %s" %s args.input_file
inputs = np.load(args.input_file)
elif os.path.isdir(args.input_file):
+ print("Loading folder: %s" % args.input_file)
inputs =[caffe.io.load_image(im_f)
for im_f in glob.glob(args.input_file + '/*.' + args.ext)]
else:
+ print("Loading file: %s" % args.input_file)
inputs = [caffe.io.load_image(args.input_file)]
print("Classifying %d inputs." % len(inputs))
print("Done in %.2f s." % (time.time() - start))
# Save
+ print("Saving results into %s" % args.output_file)
np.save(args.output_file, predictions)
# Load input.
t = time.time()
- print('Loading input...')
+ print("Loading input...")
if args.input_file.lower().endswith('txt'):
with open(args.input_file) as f:
inputs = [_.strip() for _ in f.readlines()]