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