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')