def decafImages(src_path, output_path, socketid, result_path, single_file_name=''):
try:
#Entire Directory
if os.path.isdir(os.path.join(src_path,single_file_name)):
for file_name in os.listdir(src_path):
tags = {}
image_path = os.path.join(src_path, file_name)
if os.path.isfile(image_path):
""" Trying to get the output of classify python script to send to user - Part 1/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
print 'Running caffe classify on multiple images'
mat_file_path = decaf.calculate_decaf_image(file_name, src_path, output_path, 3, socketid, tags)
print tags
""" Part 2/2
sys.stdout=old_stdout
"""
log_to_terminal("Results: "+str(tags), socketid)
# sorted_tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# webResult = {}
#webResult[str(result_path + file_name)] = sorted_tags
result_url = urlparse(result_path).path
r.publish('chat',
json.dumps({'web_result': os.path.join(result_url, 'results', file_name+'.mat'), 'socketid': str(socketid)}))
log_to_terminal('Thank you for using CloudCV', socketid)
# Single File
else:
""" Part 3/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
tags = {}
print 'Running caffe classify on a single image: ' + single_file_name
mat_file_path = decaf.calculate_decaf_image(single_file_name, src_path, output_path, 3, socketid, tags)
""" Part 4/4
sys.stdout=old_stdout
"""
log_to_terminal("Results: "+str(tags), socketid)
# tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# web_result = {}
# web_result[str(result_path)] = tags
result_url = os.path.dirname(urlparse(result_path).path)
r.publish('chat', json.dumps({'web_result': os.path.join(result_url, 'results', single_file_name+'.mat'), 'socketid': str(socketid)}))
log_to_terminal('Thank you for using CloudCV', socketid)
except Exception as e:
log_to_terminal(str(traceback.format_exc()), socketid)
def decafImages(src_path, output_path, socketid, result_path, single_file_name='', modelname=''):
if modelname is not '':
lMODEL_FILE = str(os.path.join(conf.CAFFE_DIR, 'models',modelname,'deploy.prototxt'))
lPRETRAINED = str(os.path.join(conf.CAFFE_DIR, 'models', modelname, modelname+'.caffemodel'))
r.publish('chat', json.dumps({'error': lMODEL_FILE+' '+lPRETRAINED, 'socketid': socketid}))
caffe.set_phase_test()
caffe.set_mode_cpu()
modelnet = caffe.Classifier(lMODEL_FILE, lPRETRAINED)
#r.publish('chat', json.dumps({'error': str(modelname), 'socketid': socketid}))
else:
modelnet = None
try:
#Entire Directory
if os.path.isdir(os.path.join(src_path,single_file_name)):
for file_name in os.listdir(src_path):
tags = {}
image_path = os.path.join(src_path, file_name)
if os.path.isfile(image_path):
""" Trying to get the output of classify python script to send to user - Part 1/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
print 'Running caffe classify on multiple images'
mat_file_path = decaf.calculate_decaf_image(file_name, src_path, output_path, 3, socketid, tags, modelname, modelnet)
print tags
""" Part 2/2
sys.stdout=old_stdout
"""
log_to_terminal("Results: "+str(tags), socketid)
# sorted_tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# webResult = {}
#webResult[str(result_path + file_name)] = sorted_tags
result_url = urlparse(result_path).path
r.publish('chat',
json.dumps({'web_result': os.path.join(result_url, 'results', file_name+'.mat'), 'socketid': str(socketid)}))
log_to_terminal('Thank you for using CloudCV', socketid)
# Single File
else:
""" Part 3/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
tags = {}
print 'Running caffe classify on a single image: ' + single_file_name
try:
mat_file_path = decaf.calculate_decaf_image(single_file_name, src_path, output_path, 3, socketid, tags, modelname, modelnet)
except Exception as e:
print str(e)
""" Part 4/4
sys.stdout=old_stdout
"""
log_to_terminal("Results: "+str(tags), socketid)
# tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# web_result = {}
# web_result[str(result_path)] = tags
result_url = os.path.dirname(urlparse(result_path).path)
r.publish('chat', json.dumps({'web_result': os.path.join(result_url, 'results', single_file_name+'.mat'), 'socketid': str(socketid)}))
log_to_terminal('Thank you for using CloudCV', socketid)
except Exception as e:
log_to_terminal(str(traceback.format_exc()), socketid)
def decafImages(src_path,
output_path,
socketid,
result_path,
single_file_name='',
modelname=''):
if modelname is not '':
lMODEL_FILE = str(
os.path.join(conf.CAFFE_DIR, 'models', modelname,
'deploy.prototxt'))
lPRETRAINED = str(
os.path.join(conf.CAFFE_DIR, 'models', modelname,
modelname + '.caffemodel'))
r.publish(
'chat',
json.dumps({
'error': lMODEL_FILE + ' ' + lPRETRAINED,
'socketid': socketid
}))
caffe.set_phase_test()
caffe.set_mode_cpu()
modelnet = caffe.Classifier(lMODEL_FILE, lPRETRAINED)
#r.publish('chat', json.dumps({'error': str(modelname), 'socketid': socketid}))
else:
modelnet = None
try:
#Entire Directory
if os.path.isdir(os.path.join(src_path, single_file_name)):
for file_name in os.listdir(src_path):
tags = {}
image_path = os.path.join(src_path, file_name)
if os.path.isfile(image_path):
""" Trying to get the output of classify python script to send to user - Part 1/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
print 'Running caffe classify on multiple images'
mat_file_path = decaf.calculate_decaf_image(
file_name, src_path, output_path, 3, socketid, tags,
modelname, modelnet)
print tags
""" Part 2/2
sys.stdout=old_stdout
"""
log_to_terminal("Results: " + str(tags), socketid)
# sorted_tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# webResult = {}
#webResult[str(result_path + file_name)] = sorted_tags
result_url = urlparse(result_path).path
r.publish(
'chat',
json.dumps({
'web_result':
os.path.join(result_url, 'results',
file_name + '.mat'),
'socketid':
str(socketid)
}))
log_to_terminal('Thank you for using CloudCV', socketid)
# Single File
else:
""" Part 3/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
tags = {}
print 'Running caffe classify on a single image: ' + single_file_name
try:
mat_file_path = decaf.calculate_decaf_image(
single_file_name, src_path, output_path, 3, socketid, tags,
modelname, modelnet)
except Exception as e:
print str(e)
""" Part 4/4
sys.stdout=old_stdout
"""
log_to_terminal("Results: " + str(tags), socketid)
# tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# web_result = {}
# web_result[str(result_path)] = tags
result_url = os.path.dirname(urlparse(result_path).path)
r.publish(
'chat',
json.dumps({
'web_result':
os.path.join(result_url, 'results',
single_file_name + '.mat'),
'socketid':
str(socketid)
}))
log_to_terminal('Thank you for using CloudCV', socketid)
except Exception as e:
log_to_terminal(str(traceback.format_exc()), socketid)
def decafImages(src_path,
output_path,
socketid,
result_path,
single_file_name=''):
try:
#Entire Directory
if os.path.isdir(os.path.join(src_path, single_file_name)):
for file_name in os.listdir(src_path):
tags = {}
image_path = os.path.join(src_path, file_name)
if os.path.isfile(image_path):
""" Trying to get the output of classify python script to send to user - Part 1/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
print 'Running caffe classify on multiple images'
mat_file_path = decaf.calculate_decaf_image(
file_name, src_path, output_path, 3, socketid, tags)
print tags
""" Part 2/2
sys.stdout=old_stdout
"""
log_to_terminal("Results: " + str(tags), socketid)
# sorted_tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# webResult = {}
#webResult[str(result_path + file_name)] = sorted_tags
result_url = urlparse(result_path).path
r.publish(
'chat',
json.dumps({
'web_result':
os.path.join(result_url, 'results',
file_name + '.mat'),
'socketid':
str(socketid)
}))
log_to_terminal('Thank you for using CloudCV', socketid)
# Single File
else:
""" Part 3/4
myPrint = CustomPrint(socketid)
old_stdout=sys.stdout
sys.stdout = myPrint
"""
tags = {}
print 'Running caffe classify on a single image: ' + single_file_name
mat_file_path = decaf.calculate_decaf_image(
single_file_name, src_path, output_path, 3, socketid, tags)
""" Part 4/4
sys.stdout=old_stdout
"""
log_to_terminal("Results: " + str(tags), socketid)
# tags = sorted(tags.iteritems(), key=operator.itemgetter(1), reverse=True)
# web_result = {}
# web_result[str(result_path)] = tags
result_url = os.path.dirname(urlparse(result_path).path)
r.publish(
'chat',
json.dumps({
'web_result':
os.path.join(result_url, 'results',
single_file_name + '.mat'),
'socketid':
str(socketid)
}))
log_to_terminal('Thank you for using CloudCV', socketid)
except Exception as e:
log_to_terminal(str(traceback.format_exc()), socketid)
