基于卷积神经网络的车牌识别

from re import U
import simplejson
from django.shortcuts import render
from django.contrib.auth import authenticate, login, logout
from django.contrib.auth.models import User
from django.http.request import HttpRequest
from index.utils import success, error
from functools import wraps
from django.http import HttpResponseRedirect
from django.urls import reverse
from django.http import JsonResponse
# 用户登录装饰器，用于限制必须登录的情况
def login_required(func):
    @wraps(func)
    def wrap(request: HttpRequest, *args, **kwargs):
        # 如果用户已登陆，允许访问
        if request.user.is_authenticated:
            return func(request, *args, **kwargs)
        return error(code=401)
    return wrap
# 用户登录视图
def login_view(request: HttpRequest):
    data = simplejson.loads(request.body)
    user = authenticate(
        request, username=data.get("username"), password=data.get("password")
    )
    if user:
        # 账号密码成功，进入主页面
        return success(data=user.id)
    return error("帐号或密码错误")
@login_required
def get_user_view(request: HttpRequest):
    id_ = request.headers.get("access-token")
    user = User.objects.filter(id=id_).first()
    if user:
        return success(
            data={"name": user.username, "role": [], "isSuperuser": user.is_superuser}
        )
# 用户注册视图
def signup(request):
    data = simplejson.loads(request.body)
    if not all((data.get("username"), data.get("password"), data.get("password2"))):
        return error("信息不全")
    if data.get("password") != data.get("password2"):
        return error("二次密码不一致")
    if User.objects.filter(username=data.get("username")).exists():
        return error("帐号已存在")
    user = User.objects.create_user(
        username=data.get("username"), password=data.get("password"), is_staff=True
    )  # , is_superuser=True
    # 注册新用户
    user.save()
    return success(data=user.id)
# 工具函数
def to_dict(l):
    def _todict(obj):
        j = {k: v for k, v in obj.__dict__.items() if not k.startswith("_")}
        return j
    return [_todict(i) for i in l]

from base64 import b64decode
import time
import re
import json
from django.http import HttpResponse, JsonResponse
from django.shortcuts import render
from django.core.files.base import ContentFile
from subprocess import call
import os
from django.db.models.fields.files import ImageFieldFile
from django.db.models import ImageField
from datetime import datetime
from . import detect as dt
from django.core.paginator import Paginator
from django.db.models import Q
from hashlib import md5
from .models import *
def to_dict(l, exclude=tuple()):
    # 将数据库模型 变为 字典数据 的工具类函数
    def transform(v):
        # print(type(v))
        if isinstance(v, datetime):
            return v.strftime("%Y-%m-%d %H:%M:%S")
        elif isinstance(v, ImageFieldFile):
            return f"http://127.0.0.1:8000/media/{v.url}"
        return v
    def _todict(obj):
        j = {
            k: transform(v)
            for k, v in obj.__dict__.items()
            if not k.startswith("_") and k not in exclude
        }
        return j
    return [_todict(i) for i in l]
def plate_detect(request):
    """车牌识别"""
    data = request.json
    image = data.get("image")
    ext, bcontent = re.search(r"data:image/(.*?);base64,(.*)", image).groups()
    name = str(int(time.time() * 1000)) + f".{ext}"
    content = b64decode(bcontent)
    hash_md5 = md5(content).hexdigest()
    p = Photo.objects.filter(md5=hash_md5, user=request.user).first()
    if p is None:
        result = dt(image)
        if result is None:
            result = []
        p = Photo(
            md5=hash_md5,
            user=request.user,
            result=json.dumps(result, ensure_ascii=False, indent=4),
        )
        p.image.save(name, ContentFile(content))
        p.save()
        p = Photo.objects.get(pk=p.id)
    i = to_dict([p])[0]
    i["image"] = "http://127.0.0.1:8000/media/" + i["image"]
    i["result"] = json.loads(i["result"])
    return JsonResponse(i)
def get_list(request):
    # 列表
    body = request.json
    pagesize = body.get("pagesize", 10)
    page = body.get("page", 1)
    query = {
        k[1:]: v
        for k, v in body.items()
        if k.startswith("_") and (v != "" and v is not None)
    }
    q = Q(**query)
    objs = Photo.objects.filter(q).order_by("-id")
    paginator = Paginator(objs, pagesize)
    pg = paginator.page(page)
    result = to_dict(pg.object_list)
    for i in result:
        i["image"] = "http://127.0.0.1:8000/media/" + i["image"]
        i["result"] = json.loads(i["result"])
    return JsonResponse({"total": paginator.count, "result": result})