administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-05 10:45:21 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix type checks

Browse Source
This commit is contained in:
Blake Blackshear 2023-04-23 16:49:02 -05:00
parent 4c0addeb08
commit df5d18f1ef
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
frigate/plus.py
View File

@ -3,6 +3,7 @@ import json
import logging import logging
import os import os
import re import re
from typing import List
import requests import requests
from frigate.const import PLUS_ENV_VAR, PLUS_API_HOST from frigate.const import PLUS_ENV_VAR, PLUS_API_HOST
from requests.models import Response from requests.models import Response
@ -133,7 +134,15 @@ class PlusApi:
return str(presigned_urls.get("imageId")) return str(presigned_urls.get("imageId"))
def add_false_positive( def add_false_positive(
self, plus_id, region, bbox, score, label, model_hash, model_type, detector_type self,
plus_id: str,
region: List[float],
bbox: List[float],
score: float,
label: str,
model_hash: str,
model_type: str,
detector_type: str,
) -> None: ) -> None:
r = self._put( r = self._put(
f"image/{plus_id}/false_positive", f"image/{plus_id}/false_positive",
@ -157,7 +166,13 @@ class PlusApi:
if not r.ok: if not r.ok:
raise Exception(r.text) raise Exception(r.text)
def add_annotation(self, plus_id, bbox, label, difficult=False) -> None: def add_annotation(
self,
plus_id: str,
bbox: List[float],
label: str,
difficult: bool = False,
) -> None:
r = self._put( r = self._put(
f"image/{plus_id}/annotation", f"image/{plus_id}/annotation",
{ {