def Classify(request):
try:
print 'try'
img_predict = img_classify.imgClassify()
img_dir = './tmpl/images/'
img = request.FILES.get('img')
image_name = str(datetime.datetime.now()).replace(' ', '_') + '_' + img.name
image = Image.open(img)
img_dir = img_dir + image_name
if 1:
#============= image rotate start =====
if hasattr(image, '_getexif'):
exifdata = image._getexif()
if exifdata != None:
try:
orient = exifdata[0x0112]
switch = {
'1': lambda: img_rotate(image,0),
'2': lambda: img_rotate(image,0),
'3': lambda: img_rotate(image,180),
'4': lambda: img_rotate(image,180),
'5': lambda: img_rotate(image,90),
'6': lambda: img_rotate(image,270),
'7': lambda: img_rotate(image,270),
'8': lambda: img_rotate(image,90)
}
image = switch[str(orient)]()
except:
None
#============= image rotate end =======
image.save(img_dir)
results = img_predict.classify(img_dir)
final = []
final.append(1)
final.append(image_name)
final.append(results)
print 'return'
return HttpResponse(json.dumps(final), content_type='application/jason')
except:
print 'except'
final = [0,0]
return HttpResponse(json.dumps(final), content_type='application/jason')
def imageClassify(request):
global img_predict
img_predict = img_classify.imgClassify()
return render(request, 'images.html')
