CaesarAI Fixed I think

CaesarObjectDetection/CaesarYolo.py

@@ -10,7 +10,7 @@ class CaesarYolo:
         self.CONFIDENCE = 0.5
         self.SCORE_THRESHOLD = 0.5
         self.IOU_THRESHOLD = 0.5
-        self.current_dir = os.path.realpath(__file__).replace(f"/CaesarYolo.py","")
+        self.current_dir = os.path.realpath(__file__).replace(f"\\CaesarYolo.py","")
         config_path = f"{self.current_dir}/cfg/yolov3.cfg"
         weights_path = f"{self.current_dir}/weights/yolov3.weights"
         self.font_scale = 1

RequestModels.py

@@ -51,7 +51,6 @@ class CaesarObjectDetectModel(BaseModel):
     shape:List
 class CaesarOCRHTTPModel(BaseModel):
     frame: str
-    shape:List
 
 class TriggerAPIModel(BaseModel):
     user_trigger : str

main.py

@@ -93,7 +93,9 @@ async def caesarfacedetect(websocket: WebSocket):
 def caesarfacesnap(frames: CaesarOCRHTTPModel):
     try:
         frames = dict(frames)
-        image = np.frombuffer(base64.b64decode(frames["frame"]),dtype="uint8").reshape(frames["shape"][0],frames["shape"][1],3)
+        image_arr = np.frombuffer(base64.b64decode(frames["frame"]),dtype="uint8")
+        image = cv2.imdecode(image_arr,flags=cv2.IMREAD_COLOR)
+        #.reshape(frames["shape"][0],frames["shape"][1],3)
         image = caesarfacedetectmodel.detect_face(image,snapcropface=True)
         if image == [] or image is None:
             return {"frame":"no face was detected."}

requirements.txt

@@ -1,114 +1,85 @@
 absl-py==1.4.0
 anyio==3.6.2
-arrow==1.2.3
-asgiref==3.6.0
 astunparse==1.6.3
 beautifulsoup4==4.11.2
-bidict==0.22.1
-binaryornot==0.4.4
 blis==0.7.9
-briefcase==0.3.9
 cachetools==5.3.0
 catalogue==2.0.8
 certifi==2022.12.7
 chardet==3.0.4
 charset-normalizer==3.0.1
 click==8.1.3
+colorama==0.4.6
 confection==0.0.4
-contourpy==1.0.6
-cookiecutter==2.1.1
+contourpy==1.0.7
 cycler==0.11.0
 cymem==2.0.7
-decorator==4.4.2
-deepface==0.0.78
+deepface==0.0.79
 dnspython==2.3.0
 en-core-web-sm @ https://github.com/explosion/spacy-models/releases/download/en_core_web_sm-3.5.0/en_core_web_sm-3.5.0-py3-none-any.whl
-eventlet==0.33.3
-fastapi==0.74.1
+fastapi==0.92.0
+fastapi-utils==0.2.1
 ffmpeg-python==0.2.0
 filelock==3.9.0
 fire==0.5.0
-Flask==2.2.2
-Flask-Cors==3.0.10
-Flask-JWT-Extended==4.4.4
+Flask==2.2.3
 flatbuffers==23.1.21
 fonttools==4.38.0
 future==0.18.3
 gast==0.4.0
-gbulb==0.6.4
-gdown==4.6.3
-gitdb==4.0.10
-GitPython==3.1.30
+gdown==4.6.4
 google-auth==2.16.1
 google-auth-oauthlib==0.4.6
 google-pasta==0.2.0
-googletrans==3.1.0a0
+googletrans==3.0.0
 greenlet==2.0.2
-grpcio==1.51.1
-gTTS==2.2.4
-gTTS-token==1.1.4
+grpcio==1.51.3
+gTTS==2.3.1
+gunicorn==20.1.0
 h11==0.9.0
 h2==3.2.0
 h5py==3.8.0
 hpack==3.0.0
 hstspreload==2023.1.1
 httpcore==0.9.1
-httptools==0.5.0
 httpx==0.13.3
-huggingface-hub==0.12.0
+huggingface-hub==0.12.1
 hyperframe==5.2.0
 idna==2.10
-imageio==2.25.1
-imageio-ffmpeg==0.4.8
 importlib-metadata==6.0.0
-imutils==0.5.4
-install==1.3.5
+importlib-resources==5.12.0
 itsdangerous==2.1.2
 Jinja2==3.1.2
-jinja2-time==0.2.0
 keras==2.11.0
-Kivy==2.1.0
-Kivy-examples==2.1.0
-Kivy-Garden==0.1.5
 kiwisolver==1.4.4
 langcodes==3.3.0
 libclang==15.0.6.1
 lxml==4.9.2
 Markdown==3.4.1
-markdown-it-py==2.1.0
 MarkupSafe==2.1.2
-matplotlib==3.5.3
-mdurl==0.1.2
-moviepy==1.0.2
+matplotlib==3.7.0
 mtcnn==0.1.1
 murmurhash==1.0.9
-numpy==1.21.6
+numpy==1.24.2
 oauthlib==3.2.2
-opencv-python==4.7.0.68
+opencv-python==4.7.0.72
 opt-einsum==3.3.0
 packaging==23.0
-pandas==1.3.5
+pandas==1.5.3
 pandas-datareader==0.10.0
 pathy==0.10.1
 Pillow==9.4.0
-platformdirs==3.0.0
 preshed==3.0.8
-proglog==0.1.10
 protobuf==3.19.6
-psutil==5.9.4
 pyasn1==0.4.8
 pyasn1-modules==0.2.8
-pycairo==1.23.0
-pydantic==1.10.4
-PyGObject==3.42.2
-PyJWT==2.6.0
+pydantic==1.10.5
 pymongo==4.3.3
 pyparsing==3.0.9
+PySocks==1.7.1
 pytesseract==0.3.10
 python-dateutil==2.8.2
-python-dotenv==0.21.1
-python-multipart==0.0.5
-python-slugify==8.0.0
+python-multipart==0.0.6
 pytz==2022.7.1
 PyYAML==6.0
 regex==2022.10.31
@@ -116,46 +87,37 @@ requests==2.28.2
 requests-oauthlib==1.3.1
 retina-face==0.0.13
 rfc3986==1.5.0
-rich==13.3.1
 rsa==4.9
-sentencepiece==0.1.97
 six==1.16.0
 smart-open==6.3.0
-smmap==5.0.0
 sniffio==1.3.0
-soupsieve==2.3.2.post1
+soupsieve==2.4
 spacy==3.5.0
 spacy-legacy==3.0.12
 spacy-loggers==1.0.4
 SpeechRecognition==3.9.0
+SQLAlchemy==1.4.46
 srsly==2.4.5
-starlette==0.17.1
+starlette==0.25.0
 tensorboard==2.11.2
 tensorboard-data-server==0.6.1
 tensorboard-plugin-wit==1.8.1
 tensorflow==2.11.0
 tensorflow-estimator==2.11.0
-tensorflow-io-gcs-filesystem==0.30.0
+tensorflow-intel==2.11.0
+tensorflow-io-gcs-filesystem==0.31.0
 termcolor==2.2.0
-text-unidecode==1.3
 thinc==8.1.7
-toga-core==0.3.0
-toga-gtk==0.3.0
 tokenizers==0.13.2
-tomli==2.0.1
-torch==1.11.0
-tqdm
+torch==1.13.1
+tqdm==4.64.1
 transformers==4.26.1
-travertino==0.1.3
 typer==0.7.0
-typing_extensions==4.4.0
+typing_extensions==4.5.0
 urllib3==1.26.14
-uvicorn==0.17.6
-uvloop==0.17.0
+uvicorn==0.20.0
 wasabi==1.1.1
-watchgod==0.8.2
-websockets==10.4
-Werkzeug==2.2.2
-wrapt==1.14.1
-wsproto==1.2.0
-zipp==3.13.0
+Werkzeug==2.2.3
+wincertstore==0.2
+wrapt==1.15.0
+zipp==3.15.0

test.txt

@@ -0,0 +1,27 @@
+altgraph==0.17.3
+anyio==3.6.2
+axios==0.1.0
+click==8.1.3
+colorama==0.4.6
+comtypes==1.1.14
+fastapi==0.92.0
+fastapi-utils==0.2.1
+flask-sock==0.6.0
+greenlet==2.0.2
+idna==3.4
+pefile==2022.5.30
+playsound==1.3.0
+pydantic==1.10.5
+Pygments==2.14.0
+pyinstaller==5.7.0
+pyinstaller-hooks-contrib==2022.14
+pypiwin32==223
+pyttsx3==2.90
+pywin32-ctypes==0.2.0
+resemble==1.3.0
+simple-websocket==0.9.0
+sniffio==1.3.0
+SQLAlchemy==1.4.46
+starlette==0.25.0
+typing_extensions==4.5.0
+wincertstore==0.2

test1.py

@@ -0,0 +1,6 @@
+import requests
+import json
+with open("imtest.json","r") as f:
+    tets = json.load(f)["info"]
+resp = requests.post("http://10.0.170.41:7860/caesarocr",json={"frame":tets})
+print(resp.json())