def images_results(request): if request.method == "POST": print(request.POST) form = UploadImageForm(request.POST, request.FILES) if form.is_valid(): algo = request.POST["algo"] img = Image(img=request.FILES["file"]) if os.path.exists(settings.MEDIA_ROOT + "/" + settings.UPLOAD_PATH): shutil.rmtree(settings.MEDIA_ROOT + "/" + settings.UPLOAD_PATH) img.save() img_searcher = ImageSearcher() images = img_searcher.similar_images(img, algo) return JSONResponse(images) else: return JSONResponse({"bad": "not biv"}) else: return JSONResponse({"options": "options"})
def convnet(): convnet = pickle.load(open('../pascal/best_model.p', 'rb')) images, names = self.two_labels_data() features = convnet.get_features(images) for feature, name in zip(features, names): img = Image( features=feature, name = name ) img.save()