mirror of
https://github.com/blakeblackshear/frigate.git
synced 2024-11-21 19:07:46 +01:00
False positives (#6217)
* add false positive submission * switch timeline events to x,y,w,h * update docs * fix type checks * convert to upsert * fix config test
This commit is contained in:
parent
9dca1e1d9f
commit
0d16bd0144
@ -37,9 +37,12 @@
|
|||||||
"onAutoForward": "silent"
|
"onAutoForward": "silent"
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
"customizations": {
|
||||||
|
"vscode": {
|
||||||
"extensions": [
|
"extensions": [
|
||||||
"ms-python.vscode-pylance",
|
|
||||||
"ms-python.python",
|
"ms-python.python",
|
||||||
|
"ms-python.vscode-pylance",
|
||||||
|
"ms-python.black-formatter",
|
||||||
"visualstudioexptteam.vscodeintellicode",
|
"visualstudioexptteam.vscodeintellicode",
|
||||||
"mhutchie.git-graph",
|
"mhutchie.git-graph",
|
||||||
"ms-azuretools.vscode-docker",
|
"ms-azuretools.vscode-docker",
|
||||||
@ -55,7 +58,7 @@
|
|||||||
"remote.autoForwardPorts": false,
|
"remote.autoForwardPorts": false,
|
||||||
"python.linting.pylintEnabled": true,
|
"python.linting.pylintEnabled": true,
|
||||||
"python.linting.enabled": true,
|
"python.linting.enabled": true,
|
||||||
"python.formatting.provider": "black",
|
"python.formatting.provider": "none",
|
||||||
"python.languageServer": "Pylance",
|
"python.languageServer": "Pylance",
|
||||||
"editor.formatOnPaste": false,
|
"editor.formatOnPaste": false,
|
||||||
"editor.formatOnSave": true,
|
"editor.formatOnSave": true,
|
||||||
@ -65,14 +68,23 @@
|
|||||||
"python.testing.unittestArgs": ["-v", "-s", "./frigate/test"],
|
"python.testing.unittestArgs": ["-v", "-s", "./frigate/test"],
|
||||||
"files.trimTrailingWhitespace": true,
|
"files.trimTrailingWhitespace": true,
|
||||||
"eslint.workingDirectories": ["./web"],
|
"eslint.workingDirectories": ["./web"],
|
||||||
|
"[python]": {
|
||||||
|
"editor.defaultFormatter": "ms-python.black-formatter",
|
||||||
|
"editor.formatOnSave": true
|
||||||
|
},
|
||||||
"[json][jsonc]": {
|
"[json][jsonc]": {
|
||||||
"editor.defaultFormatter": "esbenp.prettier-vscode"
|
"editor.defaultFormatter": "esbenp.prettier-vscode"
|
||||||
},
|
},
|
||||||
"[jsx][js][tsx][ts]": {
|
"[jsx][js][tsx][ts]": {
|
||||||
"editor.codeActionsOnSave": ["source.addMissingImports", "source.fixAll"],
|
"editor.codeActionsOnSave": [
|
||||||
|
"source.addMissingImports",
|
||||||
|
"source.fixAll"
|
||||||
|
],
|
||||||
"editor.tabSize": 2
|
"editor.tabSize": 2
|
||||||
},
|
},
|
||||||
"cSpell.ignoreWords": ["rtmp"],
|
"cSpell.ignoreWords": ["rtmp"],
|
||||||
"cSpell.words": ["preact"]
|
"cSpell.words": ["preact"]
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
@ -221,6 +221,7 @@ http {
|
|||||||
|
|
||||||
add_header 'Access-Control-Allow-Origin' '*';
|
add_header 'Access-Control-Allow-Origin' '*';
|
||||||
add_header 'Access-Control-Allow-Methods' 'GET, POST, PUT, DELETE, OPTIONS';
|
add_header 'Access-Control-Allow-Methods' 'GET, POST, PUT, DELETE, OPTIONS';
|
||||||
|
add_header 'Access-Control-Allow-Headers' 'DNT,User-Agent,X-Requested-With,If-Modified-Since,Cache-Control,Content-Type,Range';
|
||||||
proxy_pass http://frigate_api/;
|
proxy_pass http://frigate_api/;
|
||||||
proxy_pass_request_headers on;
|
proxy_pass_request_headers on;
|
||||||
proxy_set_header Host $host;
|
proxy_set_header Host $host;
|
||||||
|
@ -24,7 +24,6 @@ Examples of available modules are:
|
|||||||
- `frigate.app`
|
- `frigate.app`
|
||||||
- `frigate.mqtt`
|
- `frigate.mqtt`
|
||||||
- `frigate.object_detection`
|
- `frigate.object_detection`
|
||||||
- `frigate.zeroconf`
|
|
||||||
- `detector.<detector_name>`
|
- `detector.<detector_name>`
|
||||||
- `watchdog.<camera_name>`
|
- `watchdog.<camera_name>`
|
||||||
- `ffmpeg.<camera_name>.<sorted_roles>` NOTE: All FFmpeg logs are sent as `error` level.
|
- `ffmpeg.<camera_name>.<sorted_roles>` NOTE: All FFmpeg logs are sent as `error` level.
|
||||||
|
@ -173,8 +173,8 @@ Events from the database. Accepts the following query string parameters:
|
|||||||
Timeline of key moments of an event(s) from the database. Accepts the following query string parameters:
|
Timeline of key moments of an event(s) from the database. Accepts the following query string parameters:
|
||||||
|
|
||||||
| param | Type | Description |
|
| param | Type | Description |
|
||||||
| -------------------- | ---- | --------------------------------------------- |
|
| ----------- | ---- | ----------------------------------- |
|
||||||
| `camera` | int | Name of camera |
|
| `camera` | str | Name of camera |
|
||||||
| `source_id` | str | ID of tracked object |
|
| `source_id` | str | ID of tracked object |
|
||||||
| `limit` | int | Limit the number of events returned |
|
| `limit` | int | Limit the number of events returned |
|
||||||
|
|
||||||
@ -198,6 +198,14 @@ Sets retain to true for the event id.
|
|||||||
|
|
||||||
Submits the snapshot of the event to Frigate+ for labeling.
|
Submits the snapshot of the event to Frigate+ for labeling.
|
||||||
|
|
||||||
|
| param | Type | Description |
|
||||||
|
| -------------------- | ---- | ---------------------------------- |
|
||||||
|
| `include_annotation` | int | Submit annotation to Frigate+ too. |
|
||||||
|
|
||||||
|
### `PUT /api/events/<id>/false_positive`
|
||||||
|
|
||||||
|
Submits the snapshot of the event to Frigate+ for labeling and adds the detection as a false positive.
|
||||||
|
|
||||||
### `DELETE /api/events/<id>/retain`
|
### `DELETE /api/events/<id>/retain`
|
||||||
|
|
||||||
Sets retain to false for the event id (event may be deleted quickly after removing).
|
Sets retain to false for the event id (event may be deleted quickly after removing).
|
||||||
|
@ -18,6 +18,7 @@ from frigate.const import (
|
|||||||
REGEX_CAMERA_NAME,
|
REGEX_CAMERA_NAME,
|
||||||
YAML_EXT,
|
YAML_EXT,
|
||||||
)
|
)
|
||||||
|
from frigate.detectors.detector_config import BaseDetectorConfig
|
||||||
from frigate.util import (
|
from frigate.util import (
|
||||||
create_mask,
|
create_mask,
|
||||||
deep_merge,
|
deep_merge,
|
||||||
@ -770,7 +771,7 @@ def verify_config_roles(camera_config: CameraConfig) -> None:
|
|||||||
|
|
||||||
def verify_valid_live_stream_name(
|
def verify_valid_live_stream_name(
|
||||||
frigate_config: FrigateConfig, camera_config: CameraConfig
|
frigate_config: FrigateConfig, camera_config: CameraConfig
|
||||||
) -> None:
|
) -> ValueError | None:
|
||||||
"""Verify that a restream exists to use for live view."""
|
"""Verify that a restream exists to use for live view."""
|
||||||
if (
|
if (
|
||||||
camera_config.live.stream_name
|
camera_config.live.stream_name
|
||||||
@ -848,7 +849,7 @@ class FrigateConfig(FrigateBaseModel):
|
|||||||
model: ModelConfig = Field(
|
model: ModelConfig = Field(
|
||||||
default_factory=ModelConfig, title="Detection model configuration."
|
default_factory=ModelConfig, title="Detection model configuration."
|
||||||
)
|
)
|
||||||
detectors: Dict[str, DetectorConfig] = Field(
|
detectors: Dict[str, BaseDetectorConfig] = Field(
|
||||||
default=DEFAULT_DETECTORS,
|
default=DEFAULT_DETECTORS,
|
||||||
title="Detector hardware configuration.",
|
title="Detector hardware configuration.",
|
||||||
)
|
)
|
||||||
@ -1031,7 +1032,15 @@ class FrigateConfig(FrigateBaseModel):
|
|||||||
detector_config.model.dict(exclude_unset=True),
|
detector_config.model.dict(exclude_unset=True),
|
||||||
config.model.dict(exclude_unset=True),
|
config.model.dict(exclude_unset=True),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
if not "path" in merged_model:
|
||||||
|
if detector_config.type == "cpu":
|
||||||
|
merged_model["path"] = "/cpu_model.tflite"
|
||||||
|
elif detector_config.type == "edgetpu":
|
||||||
|
merged_model["path"] = "/edgetpu_model.tflite"
|
||||||
|
|
||||||
detector_config.model = ModelConfig.parse_obj(merged_model)
|
detector_config.model = ModelConfig.parse_obj(merged_model)
|
||||||
|
detector_config.model.compute_model_hash()
|
||||||
config.detectors[key] = detector_config
|
config.detectors[key] = detector_config
|
||||||
|
|
||||||
return config
|
return config
|
||||||
|
@ -1,3 +1,4 @@
|
|||||||
|
import hashlib
|
||||||
import logging
|
import logging
|
||||||
from enum import Enum
|
from enum import Enum
|
||||||
from typing import Dict, List, Optional, Tuple, Union, Literal
|
from typing import Dict, List, Optional, Tuple, Union, Literal
|
||||||
@ -49,6 +50,7 @@ class ModelConfig(BaseModel):
|
|||||||
)
|
)
|
||||||
_merged_labelmap: Optional[Dict[int, str]] = PrivateAttr()
|
_merged_labelmap: Optional[Dict[int, str]] = PrivateAttr()
|
||||||
_colormap: Dict[int, Tuple[int, int, int]] = PrivateAttr()
|
_colormap: Dict[int, Tuple[int, int, int]] = PrivateAttr()
|
||||||
|
_model_hash: str = PrivateAttr()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def merged_labelmap(self) -> Dict[int, str]:
|
def merged_labelmap(self) -> Dict[int, str]:
|
||||||
@ -58,6 +60,10 @@ class ModelConfig(BaseModel):
|
|||||||
def colormap(self) -> Dict[int, Tuple[int, int, int]]:
|
def colormap(self) -> Dict[int, Tuple[int, int, int]]:
|
||||||
return self._colormap
|
return self._colormap
|
||||||
|
|
||||||
|
@property
|
||||||
|
def model_hash(self) -> str:
|
||||||
|
return self._model_hash
|
||||||
|
|
||||||
def __init__(self, **config):
|
def __init__(self, **config):
|
||||||
super().__init__(**config)
|
super().__init__(**config)
|
||||||
|
|
||||||
@ -67,6 +73,13 @@ class ModelConfig(BaseModel):
|
|||||||
}
|
}
|
||||||
self._colormap = {}
|
self._colormap = {}
|
||||||
|
|
||||||
|
def compute_model_hash(self) -> None:
|
||||||
|
with open(self.path, "rb") as f:
|
||||||
|
file_hash = hashlib.md5()
|
||||||
|
while chunk := f.read(8192):
|
||||||
|
file_hash.update(chunk)
|
||||||
|
self._model_hash = file_hash.hexdigest()
|
||||||
|
|
||||||
def create_colormap(self, enabled_labels: set[str]) -> None:
|
def create_colormap(self, enabled_labels: set[str]) -> None:
|
||||||
"""Get a list of colors for enabled labels."""
|
"""Get a list of colors for enabled labels."""
|
||||||
cmap = plt.cm.get_cmap("tab10", len(enabled_labels))
|
cmap = plt.cm.get_cmap("tab10", len(enabled_labels))
|
||||||
|
@ -27,7 +27,7 @@ class CpuTfl(DetectionApi):
|
|||||||
|
|
||||||
def __init__(self, detector_config: CpuDetectorConfig):
|
def __init__(self, detector_config: CpuDetectorConfig):
|
||||||
self.interpreter = Interpreter(
|
self.interpreter = Interpreter(
|
||||||
model_path=detector_config.model.path or "/cpu_model.tflite",
|
model_path=detector_config.model.path,
|
||||||
num_threads=detector_config.num_threads or 3,
|
num_threads=detector_config.num_threads or 3,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -37,7 +37,7 @@ class EdgeTpuTfl(DetectionApi):
|
|||||||
edge_tpu_delegate = load_delegate("libedgetpu.so.1.0", device_config)
|
edge_tpu_delegate = load_delegate("libedgetpu.so.1.0", device_config)
|
||||||
logger.info("TPU found")
|
logger.info("TPU found")
|
||||||
self.interpreter = Interpreter(
|
self.interpreter = Interpreter(
|
||||||
model_path=detector_config.model.path or "/edgetpu_model.tflite",
|
model_path=detector_config.model.path,
|
||||||
experimental_delegates=[edge_tpu_delegate],
|
experimental_delegates=[edge_tpu_delegate],
|
||||||
)
|
)
|
||||||
except ValueError:
|
except ValueError:
|
||||||
|
@ -12,6 +12,7 @@ from frigate.const import CLIPS_DIR
|
|||||||
from frigate.models import Event
|
from frigate.models import Event
|
||||||
from frigate.timeline import TimelineSourceEnum
|
from frigate.timeline import TimelineSourceEnum
|
||||||
from frigate.types import CameraMetricsTypes
|
from frigate.types import CameraMetricsTypes
|
||||||
|
from frigate.util import to_relative_box
|
||||||
|
|
||||||
from multiprocessing.queues import Queue
|
from multiprocessing.queues import Queue
|
||||||
from multiprocessing.synchronize import Event as MpEvent
|
from multiprocessing.synchronize import Event as MpEvent
|
||||||
@ -20,22 +21,18 @@ from typing import Dict
|
|||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def should_insert_db(prev_event: Event, current_event: Event) -> bool:
|
|
||||||
"""If current event has new clip or snapshot."""
|
|
||||||
return (not prev_event["has_clip"] and not prev_event["has_snapshot"]) and (
|
|
||||||
current_event["has_clip"] or current_event["has_snapshot"]
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def should_update_db(prev_event: Event, current_event: Event) -> bool:
|
def should_update_db(prev_event: Event, current_event: Event) -> bool:
|
||||||
"""If current_event has updated fields and (clip or snapshot)."""
|
"""If current_event has updated fields and (clip or snapshot)."""
|
||||||
if current_event["has_clip"] or current_event["has_snapshot"]:
|
if current_event["has_clip"] or current_event["has_snapshot"]:
|
||||||
|
# if this is the first time has_clip or has_snapshot turned true
|
||||||
|
if not prev_event["has_clip"] and not prev_event["has_snapshot"]:
|
||||||
|
return True
|
||||||
|
# or if any of the following values changed
|
||||||
if (
|
if (
|
||||||
prev_event["top_score"] != current_event["top_score"]
|
prev_event["top_score"] != current_event["top_score"]
|
||||||
or prev_event["entered_zones"] != current_event["entered_zones"]
|
or prev_event["entered_zones"] != current_event["entered_zones"]
|
||||||
or prev_event["thumbnail"] != current_event["thumbnail"]
|
or prev_event["thumbnail"] != current_event["thumbnail"]
|
||||||
or prev_event["has_clip"] != current_event["has_clip"]
|
or prev_event["end_time"] != current_event["end_time"]
|
||||||
or prev_event["has_snapshot"] != current_event["has_snapshot"]
|
|
||||||
):
|
):
|
||||||
return True
|
return True
|
||||||
return False
|
return False
|
||||||
@ -85,81 +82,91 @@ class EventProcessor(threading.Thread):
|
|||||||
)
|
)
|
||||||
)
|
)
|
||||||
|
|
||||||
event_config: EventsConfig = self.config.cameras[camera].record.events
|
# if this is the first message, just store it and continue, its not time to insert it in the db
|
||||||
|
|
||||||
if event_type == "start":
|
if event_type == "start":
|
||||||
self.events_in_process[event_data["id"]] = event_data
|
self.events_in_process[event_data["id"]] = event_data
|
||||||
|
continue
|
||||||
|
|
||||||
elif event_type == "update" and should_insert_db(
|
if should_update_db(self.events_in_process[event_data["id"]], event_data):
|
||||||
self.events_in_process[event_data["id"]], event_data
|
camera_config = self.config.cameras[camera]
|
||||||
):
|
event_config: EventsConfig = camera_config.record.events
|
||||||
|
width = camera_config.detect.width
|
||||||
|
height = camera_config.detect.height
|
||||||
|
first_detector = list(self.config.detectors.values())[0]
|
||||||
|
|
||||||
|
start_time = event_data["start_time"] - event_config.pre_capture
|
||||||
|
end_time = (
|
||||||
|
None
|
||||||
|
if event_data["end_time"] is None
|
||||||
|
else event_data["end_time"] + event_config.post_capture
|
||||||
|
)
|
||||||
|
# score of the snapshot
|
||||||
|
score = (
|
||||||
|
None
|
||||||
|
if event_data["snapshot"] is None
|
||||||
|
else event_data["snapshot"]["score"]
|
||||||
|
)
|
||||||
|
# detection region in the snapshot
|
||||||
|
region = (
|
||||||
|
None
|
||||||
|
if event_data["snapshot"] is None
|
||||||
|
else to_relative_box(
|
||||||
|
width,
|
||||||
|
height,
|
||||||
|
event_data["snapshot"]["region"],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
# bounding box for the snapshot
|
||||||
|
box = (
|
||||||
|
None
|
||||||
|
if event_data["snapshot"] is None
|
||||||
|
else to_relative_box(
|
||||||
|
width,
|
||||||
|
height,
|
||||||
|
event_data["snapshot"]["box"],
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
# keep these from being set back to false because the event
|
||||||
|
# may have started while recordings and snapshots were enabled
|
||||||
|
# this would be an issue for long running events
|
||||||
|
if self.events_in_process[event_data["id"]]["has_clip"]:
|
||||||
|
event_data["has_clip"] = True
|
||||||
|
if self.events_in_process[event_data["id"]]["has_snapshot"]:
|
||||||
|
event_data["has_snapshot"] = True
|
||||||
|
|
||||||
|
event = {
|
||||||
|
Event.id: event_data["id"],
|
||||||
|
Event.label: event_data["label"],
|
||||||
|
Event.camera: camera,
|
||||||
|
Event.start_time: start_time,
|
||||||
|
Event.end_time: end_time,
|
||||||
|
Event.top_score: event_data["top_score"],
|
||||||
|
Event.score: score,
|
||||||
|
Event.zones: list(event_data["entered_zones"]),
|
||||||
|
Event.thumbnail: event_data["thumbnail"],
|
||||||
|
Event.region: region,
|
||||||
|
Event.box: box,
|
||||||
|
Event.has_clip: event_data["has_clip"],
|
||||||
|
Event.has_snapshot: event_data["has_snapshot"],
|
||||||
|
Event.model_hash: first_detector.model.model_hash,
|
||||||
|
Event.model_type: first_detector.model.model_type,
|
||||||
|
Event.detector_type: first_detector.type,
|
||||||
|
}
|
||||||
|
|
||||||
|
(
|
||||||
|
Event.insert(event)
|
||||||
|
.on_conflict(
|
||||||
|
conflict_target=[Event.id],
|
||||||
|
update=event,
|
||||||
|
)
|
||||||
|
.execute()
|
||||||
|
)
|
||||||
|
|
||||||
|
# update the stored copy for comparison on future update messages
|
||||||
self.events_in_process[event_data["id"]] = event_data
|
self.events_in_process[event_data["id"]] = event_data
|
||||||
# TODO: this will generate a lot of db activity possibly
|
|
||||||
Event.insert(
|
|
||||||
id=event_data["id"],
|
|
||||||
label=event_data["label"],
|
|
||||||
camera=camera,
|
|
||||||
start_time=event_data["start_time"] - event_config.pre_capture,
|
|
||||||
end_time=None,
|
|
||||||
top_score=event_data["top_score"],
|
|
||||||
false_positive=event_data["false_positive"],
|
|
||||||
zones=list(event_data["entered_zones"]),
|
|
||||||
thumbnail=event_data["thumbnail"],
|
|
||||||
region=event_data["region"],
|
|
||||||
box=event_data["box"],
|
|
||||||
area=event_data["area"],
|
|
||||||
has_clip=event_data["has_clip"],
|
|
||||||
has_snapshot=event_data["has_snapshot"],
|
|
||||||
).execute()
|
|
||||||
|
|
||||||
elif event_type == "update" and should_update_db(
|
|
||||||
self.events_in_process[event_data["id"]], event_data
|
|
||||||
):
|
|
||||||
self.events_in_process[event_data["id"]] = event_data
|
|
||||||
# TODO: this will generate a lot of db activity possibly
|
|
||||||
Event.update(
|
|
||||||
label=event_data["label"],
|
|
||||||
camera=camera,
|
|
||||||
start_time=event_data["start_time"] - event_config.pre_capture,
|
|
||||||
end_time=None,
|
|
||||||
top_score=event_data["top_score"],
|
|
||||||
false_positive=event_data["false_positive"],
|
|
||||||
zones=list(event_data["entered_zones"]),
|
|
||||||
thumbnail=event_data["thumbnail"],
|
|
||||||
region=event_data["region"],
|
|
||||||
box=event_data["box"],
|
|
||||||
area=event_data["area"],
|
|
||||||
ratio=event_data["ratio"],
|
|
||||||
has_clip=event_data["has_clip"],
|
|
||||||
has_snapshot=event_data["has_snapshot"],
|
|
||||||
).where(Event.id == event_data["id"]).execute()
|
|
||||||
|
|
||||||
elif event_type == "end":
|
|
||||||
if event_data["has_clip"] or event_data["has_snapshot"]:
|
|
||||||
# Full update for valid end of event
|
|
||||||
Event.update(
|
|
||||||
label=event_data["label"],
|
|
||||||
camera=camera,
|
|
||||||
start_time=event_data["start_time"] - event_config.pre_capture,
|
|
||||||
end_time=event_data["end_time"] + event_config.post_capture,
|
|
||||||
top_score=event_data["top_score"],
|
|
||||||
false_positive=event_data["false_positive"],
|
|
||||||
zones=list(event_data["entered_zones"]),
|
|
||||||
thumbnail=event_data["thumbnail"],
|
|
||||||
region=event_data["region"],
|
|
||||||
box=event_data["box"],
|
|
||||||
area=event_data["area"],
|
|
||||||
ratio=event_data["ratio"],
|
|
||||||
has_clip=event_data["has_clip"],
|
|
||||||
has_snapshot=event_data["has_snapshot"],
|
|
||||||
).where(Event.id == event_data["id"]).execute()
|
|
||||||
else:
|
|
||||||
# Event ended after clip & snapshot disabled,
|
|
||||||
# only end time should be updated.
|
|
||||||
Event.update(
|
|
||||||
end_time=event_data["end_time"] + event_config.post_capture
|
|
||||||
).where(Event.id == event_data["id"]).execute()
|
|
||||||
|
|
||||||
|
if event_type == "end":
|
||||||
del self.events_in_process[event_data["id"]]
|
del self.events_in_process[event_data["id"]]
|
||||||
self.event_processed_queue.put((event_data["id"], camera))
|
self.event_processed_queue.put((event_data["id"], camera))
|
||||||
|
|
||||||
|
101
frigate/http.py
101
frigate/http.py
@ -35,6 +35,7 @@ from frigate.config import FrigateConfig
|
|||||||
from frigate.const import CLIPS_DIR, MAX_SEGMENT_DURATION, RECORD_DIR
|
from frigate.const import CLIPS_DIR, MAX_SEGMENT_DURATION, RECORD_DIR
|
||||||
from frigate.models import Event, Recordings, Timeline
|
from frigate.models import Event, Recordings, Timeline
|
||||||
from frigate.object_processing import TrackedObject
|
from frigate.object_processing import TrackedObject
|
||||||
|
from frigate.plus import PlusApi
|
||||||
from frigate.stats import stats_snapshot
|
from frigate.stats import stats_snapshot
|
||||||
from frigate.util import (
|
from frigate.util import (
|
||||||
clean_camera_user_pass,
|
clean_camera_user_pass,
|
||||||
@ -42,6 +43,7 @@ from frigate.util import (
|
|||||||
restart_frigate,
|
restart_frigate,
|
||||||
vainfo_hwaccel,
|
vainfo_hwaccel,
|
||||||
get_tz_modifiers,
|
get_tz_modifiers,
|
||||||
|
to_relative_box,
|
||||||
)
|
)
|
||||||
from frigate.storage import StorageMaintainer
|
from frigate.storage import StorageMaintainer
|
||||||
from frigate.version import VERSION
|
from frigate.version import VERSION
|
||||||
@ -57,7 +59,7 @@ def create_app(
|
|||||||
stats_tracking,
|
stats_tracking,
|
||||||
detected_frames_processor,
|
detected_frames_processor,
|
||||||
storage_maintainer: StorageMaintainer,
|
storage_maintainer: StorageMaintainer,
|
||||||
plus_api,
|
plus_api: PlusApi,
|
||||||
):
|
):
|
||||||
app = Flask(__name__)
|
app = Flask(__name__)
|
||||||
|
|
||||||
@ -179,6 +181,10 @@ def send_to_plus(id):
|
|||||||
400,
|
400,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
include_annotation = (
|
||||||
|
request.json.get("include_annotation") if request.is_json else None
|
||||||
|
)
|
||||||
|
|
||||||
try:
|
try:
|
||||||
event = Event.get(Event.id == id)
|
event = Event.get(Event.id == id)
|
||||||
except DoesNotExist:
|
except DoesNotExist:
|
||||||
@ -186,6 +192,10 @@ def send_to_plus(id):
|
|||||||
logger.error(message)
|
logger.error(message)
|
||||||
return make_response(jsonify({"success": False, "message": message}), 404)
|
return make_response(jsonify({"success": False, "message": message}), 404)
|
||||||
|
|
||||||
|
# events from before the conversion to relative dimensions cant include annotations
|
||||||
|
if any(d > 1 for d in event.box):
|
||||||
|
include_annotation = None
|
||||||
|
|
||||||
if event.end_time is None:
|
if event.end_time is None:
|
||||||
logger.error(f"Unable to load clean png for in-progress event: {event.id}")
|
logger.error(f"Unable to load clean png for in-progress event: {event.id}")
|
||||||
return make_response(
|
return make_response(
|
||||||
@ -238,9 +248,96 @@ def send_to_plus(id):
|
|||||||
event.plus_id = plus_id
|
event.plus_id = plus_id
|
||||||
event.save()
|
event.save()
|
||||||
|
|
||||||
|
if not include_annotation is None:
|
||||||
|
region = event.region
|
||||||
|
box = event.box
|
||||||
|
|
||||||
|
try:
|
||||||
|
current_app.plus_api.add_annotation(
|
||||||
|
event.plus_id,
|
||||||
|
box,
|
||||||
|
event.label,
|
||||||
|
)
|
||||||
|
except Exception as ex:
|
||||||
|
logger.exception(ex)
|
||||||
|
return make_response(
|
||||||
|
jsonify({"success": False, "message": str(ex)}),
|
||||||
|
400,
|
||||||
|
)
|
||||||
|
|
||||||
return make_response(jsonify({"success": True, "plus_id": plus_id}), 200)
|
return make_response(jsonify({"success": True, "plus_id": plus_id}), 200)
|
||||||
|
|
||||||
|
|
||||||
|
@bp.route("/events/<id>/false_positive", methods=("PUT",))
|
||||||
|
def false_positive(id):
|
||||||
|
if not current_app.plus_api.is_active():
|
||||||
|
message = "PLUS_API_KEY environment variable is not set"
|
||||||
|
logger.error(message)
|
||||||
|
return make_response(
|
||||||
|
jsonify(
|
||||||
|
{
|
||||||
|
"success": False,
|
||||||
|
"message": message,
|
||||||
|
}
|
||||||
|
),
|
||||||
|
400,
|
||||||
|
)
|
||||||
|
|
||||||
|
try:
|
||||||
|
event = Event.get(Event.id == id)
|
||||||
|
except DoesNotExist:
|
||||||
|
message = f"Event {id} not found"
|
||||||
|
logger.error(message)
|
||||||
|
return make_response(jsonify({"success": False, "message": message}), 404)
|
||||||
|
|
||||||
|
# events from before the conversion to relative dimensions cant include annotations
|
||||||
|
if any(d > 1 for d in event.box):
|
||||||
|
message = f"Events prior to 0.13 cannot be submitted as false positives"
|
||||||
|
logger.error(message)
|
||||||
|
return make_response(jsonify({"success": False, "message": message}), 400)
|
||||||
|
|
||||||
|
if event.false_positive:
|
||||||
|
message = f"False positive already submitted to Frigate+"
|
||||||
|
logger.error(message)
|
||||||
|
return make_response(jsonify({"success": False, "message": message}), 400)
|
||||||
|
|
||||||
|
if not event.plus_id:
|
||||||
|
plus_response = send_to_plus(id)
|
||||||
|
if plus_response.status_code != 200:
|
||||||
|
return plus_response
|
||||||
|
# need to refetch the event now that it has a plus_id
|
||||||
|
event = Event.get(Event.id == id)
|
||||||
|
|
||||||
|
region = event.region
|
||||||
|
box = event.box
|
||||||
|
|
||||||
|
# provide top score if score is unavailable
|
||||||
|
score = event.top_score if event.score is None else event.score
|
||||||
|
|
||||||
|
try:
|
||||||
|
current_app.plus_api.add_false_positive(
|
||||||
|
event.plus_id,
|
||||||
|
region,
|
||||||
|
box,
|
||||||
|
score,
|
||||||
|
event.label,
|
||||||
|
event.model_hash,
|
||||||
|
event.model_type,
|
||||||
|
event.detector_type,
|
||||||
|
)
|
||||||
|
except Exception as ex:
|
||||||
|
logger.exception(ex)
|
||||||
|
return make_response(
|
||||||
|
jsonify({"success": False, "message": str(ex)}),
|
||||||
|
400,
|
||||||
|
)
|
||||||
|
|
||||||
|
event.false_positive = True
|
||||||
|
event.save()
|
||||||
|
|
||||||
|
return make_response(jsonify({"success": True, "plus_id": event.plus_id}), 200)
|
||||||
|
|
||||||
|
|
||||||
@bp.route("/events/<id>/retain", methods=("DELETE",))
|
@bp.route("/events/<id>/retain", methods=("DELETE",))
|
||||||
def delete_retain(id):
|
def delete_retain(id):
|
||||||
try:
|
try:
|
||||||
@ -654,6 +751,8 @@ def events():
|
|||||||
Event.retain_indefinitely,
|
Event.retain_indefinitely,
|
||||||
Event.sub_label,
|
Event.sub_label,
|
||||||
Event.top_score,
|
Event.top_score,
|
||||||
|
Event.false_positive,
|
||||||
|
Event.box,
|
||||||
]
|
]
|
||||||
|
|
||||||
if camera != "all":
|
if camera != "all":
|
||||||
|
@ -19,6 +19,7 @@ class Event(Model): # type: ignore[misc]
|
|||||||
start_time = DateTimeField()
|
start_time = DateTimeField()
|
||||||
end_time = DateTimeField()
|
end_time = DateTimeField()
|
||||||
top_score = FloatField()
|
top_score = FloatField()
|
||||||
|
score = FloatField()
|
||||||
false_positive = BooleanField()
|
false_positive = BooleanField()
|
||||||
zones = JSONField()
|
zones = JSONField()
|
||||||
thumbnail = TextField()
|
thumbnail = TextField()
|
||||||
@ -30,6 +31,9 @@ class Event(Model): # type: ignore[misc]
|
|||||||
retain_indefinitely = BooleanField(default=False)
|
retain_indefinitely = BooleanField(default=False)
|
||||||
ratio = FloatField(default=1.0)
|
ratio = FloatField(default=1.0)
|
||||||
plus_id = CharField(max_length=30)
|
plus_id = CharField(max_length=30)
|
||||||
|
model_hash = CharField(max_length=32)
|
||||||
|
detector_type = CharField(max_length=32)
|
||||||
|
model_type = CharField(max_length=32)
|
||||||
|
|
||||||
|
|
||||||
class Timeline(Model): # type: ignore[misc]
|
class Timeline(Model): # type: ignore[misc]
|
||||||
|
@ -185,7 +185,7 @@ class TrackedObject:
|
|||||||
"id": self.obj_data["id"],
|
"id": self.obj_data["id"],
|
||||||
"camera": self.camera,
|
"camera": self.camera,
|
||||||
"frame_time": self.obj_data["frame_time"],
|
"frame_time": self.obj_data["frame_time"],
|
||||||
"snapshot_time": snapshot_time,
|
"snapshot": self.thumbnail_data,
|
||||||
"label": self.obj_data["label"],
|
"label": self.obj_data["label"],
|
||||||
"sub_label": self.obj_data.get("sub_label"),
|
"sub_label": self.obj_data.get("sub_label"),
|
||||||
"top_score": self.top_score,
|
"top_score": self.top_score,
|
||||||
|
@ -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
|
||||||
@ -79,6 +80,13 @@ class PlusApi:
|
|||||||
json=data,
|
json=data,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def _put(self, path: str, data: dict) -> Response:
|
||||||
|
return requests.put(
|
||||||
|
f"{self.host}/v1/{path}",
|
||||||
|
headers=self._get_authorization_header(),
|
||||||
|
json=data,
|
||||||
|
)
|
||||||
|
|
||||||
def is_active(self) -> bool:
|
def is_active(self) -> bool:
|
||||||
return self._is_active
|
return self._is_active
|
||||||
|
|
||||||
@ -124,3 +132,58 @@ class PlusApi:
|
|||||||
|
|
||||||
# return image id
|
# return image id
|
||||||
return str(presigned_urls.get("imageId"))
|
return str(presigned_urls.get("imageId"))
|
||||||
|
|
||||||
|
def add_false_positive(
|
||||||
|
self,
|
||||||
|
plus_id: str,
|
||||||
|
region: List[float],
|
||||||
|
bbox: List[float],
|
||||||
|
score: float,
|
||||||
|
label: str,
|
||||||
|
model_hash: str,
|
||||||
|
model_type: str,
|
||||||
|
detector_type: str,
|
||||||
|
) -> None:
|
||||||
|
r = self._put(
|
||||||
|
f"image/{plus_id}/false_positive",
|
||||||
|
{
|
||||||
|
"label": label,
|
||||||
|
"x": bbox[0],
|
||||||
|
"y": bbox[1],
|
||||||
|
"w": bbox[2],
|
||||||
|
"h": bbox[3],
|
||||||
|
"regionX": region[0],
|
||||||
|
"regionY": region[1],
|
||||||
|
"regionW": region[2],
|
||||||
|
"regionH": region[3],
|
||||||
|
"score": score,
|
||||||
|
"model_hash": model_hash,
|
||||||
|
"model_type": model_type,
|
||||||
|
"detector_type": detector_type,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
if not r.ok:
|
||||||
|
raise Exception(r.text)
|
||||||
|
|
||||||
|
def add_annotation(
|
||||||
|
self,
|
||||||
|
plus_id: str,
|
||||||
|
bbox: List[float],
|
||||||
|
label: str,
|
||||||
|
difficult: bool = False,
|
||||||
|
) -> None:
|
||||||
|
r = self._put(
|
||||||
|
f"image/{plus_id}/annotation",
|
||||||
|
{
|
||||||
|
"label": label,
|
||||||
|
"x": bbox[0],
|
||||||
|
"y": bbox[1],
|
||||||
|
"w": bbox[2],
|
||||||
|
"h": bbox[3],
|
||||||
|
"difficult": difficult,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
if not r.ok:
|
||||||
|
raise Exception(r.text)
|
||||||
|
@ -54,7 +54,8 @@ class TestConfig(unittest.TestCase):
|
|||||||
"type": "openvino",
|
"type": "openvino",
|
||||||
},
|
},
|
||||||
},
|
},
|
||||||
"model": {"path": "/default.tflite", "width": 512},
|
# needs to be a file that will exist, doesnt matter what
|
||||||
|
"model": {"path": "/etc/hosts", "width": 512},
|
||||||
}
|
}
|
||||||
|
|
||||||
frigate_config = FrigateConfig(**(deep_merge(config, self.minimal)))
|
frigate_config = FrigateConfig(**(deep_merge(config, self.minimal)))
|
||||||
@ -72,10 +73,10 @@ class TestConfig(unittest.TestCase):
|
|||||||
assert runtime_config.detectors["edgetpu"].device is None
|
assert runtime_config.detectors["edgetpu"].device is None
|
||||||
assert runtime_config.detectors["openvino"].device is None
|
assert runtime_config.detectors["openvino"].device is None
|
||||||
|
|
||||||
assert runtime_config.model.path == "/default.tflite"
|
assert runtime_config.model.path == "/etc/hosts"
|
||||||
assert runtime_config.detectors["cpu"].model.path == "/cpu_model.tflite"
|
assert runtime_config.detectors["cpu"].model.path == "/cpu_model.tflite"
|
||||||
assert runtime_config.detectors["edgetpu"].model.path == "/edgetpu_model.tflite"
|
assert runtime_config.detectors["edgetpu"].model.path == "/edgetpu_model.tflite"
|
||||||
assert runtime_config.detectors["openvino"].model.path == "/default.tflite"
|
assert runtime_config.detectors["openvino"].model.path == "/etc/hosts"
|
||||||
|
|
||||||
assert runtime_config.model.width == 512
|
assert runtime_config.model.width == 512
|
||||||
assert runtime_config.detectors["cpu"].model.width == 512
|
assert runtime_config.detectors["cpu"].model.width == 512
|
||||||
|
@ -12,6 +12,8 @@ from frigate.models import Timeline
|
|||||||
from multiprocessing.queues import Queue
|
from multiprocessing.queues import Queue
|
||||||
from multiprocessing.synchronize import Event as MpEvent
|
from multiprocessing.synchronize import Event as MpEvent
|
||||||
|
|
||||||
|
from frigate.util import to_relative_box
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
@ -64,77 +66,36 @@ class TimelineProcessor(threading.Thread):
|
|||||||
"""Handle object detection."""
|
"""Handle object detection."""
|
||||||
camera_config = self.config.cameras[camera]
|
camera_config = self.config.cameras[camera]
|
||||||
|
|
||||||
if event_type == "start":
|
timeline_entry = {
|
||||||
Timeline.insert(
|
Timeline.timestamp: event_data["frame_time"],
|
||||||
timestamp=event_data["frame_time"],
|
Timeline.camera: camera,
|
||||||
camera=camera,
|
Timeline.source: "tracked_object",
|
||||||
source="tracked_object",
|
Timeline.source_id: event_data["id"],
|
||||||
source_id=event_data["id"],
|
Timeline.data: {
|
||||||
class_type="visible",
|
"box": to_relative_box(
|
||||||
data={
|
camera_config.detect.width,
|
||||||
"box": [
|
camera_config.detect.height,
|
||||||
event_data["box"][0] / camera_config.detect.width,
|
event_data["box"],
|
||||||
event_data["box"][1] / camera_config.detect.height,
|
),
|
||||||
event_data["box"][2] / camera_config.detect.width,
|
|
||||||
event_data["box"][3] / camera_config.detect.height,
|
|
||||||
],
|
|
||||||
"label": event_data["label"],
|
"label": event_data["label"],
|
||||||
"region": [
|
"region": to_relative_box(
|
||||||
event_data["region"][0] / camera_config.detect.width,
|
camera_config.detect.width,
|
||||||
event_data["region"][1] / camera_config.detect.height,
|
camera_config.detect.height,
|
||||||
event_data["region"][2] / camera_config.detect.width,
|
event_data["region"],
|
||||||
event_data["region"][3] / camera_config.detect.height,
|
),
|
||||||
],
|
|
||||||
},
|
},
|
||||||
).execute()
|
}
|
||||||
|
if event_type == "start":
|
||||||
|
timeline_entry[Timeline.class_type] = "visible"
|
||||||
|
Timeline.insert(timeline_entry).execute()
|
||||||
elif (
|
elif (
|
||||||
event_type == "update"
|
event_type == "update"
|
||||||
and prev_event_data["current_zones"] != event_data["current_zones"]
|
and prev_event_data["current_zones"] != event_data["current_zones"]
|
||||||
and len(event_data["current_zones"]) > 0
|
and len(event_data["current_zones"]) > 0
|
||||||
):
|
):
|
||||||
Timeline.insert(
|
timeline_entry[Timeline.class_type] = "entered_zone"
|
||||||
timestamp=event_data["frame_time"],
|
timeline_entry[Timeline.data]["zones"] = event_data["current_zones"]
|
||||||
camera=camera,
|
Timeline.insert(timeline_entry).execute()
|
||||||
source="tracked_object",
|
|
||||||
source_id=event_data["id"],
|
|
||||||
class_type="entered_zone",
|
|
||||||
data={
|
|
||||||
"box": [
|
|
||||||
event_data["box"][0] / camera_config.detect.width,
|
|
||||||
event_data["box"][1] / camera_config.detect.height,
|
|
||||||
event_data["box"][2] / camera_config.detect.width,
|
|
||||||
event_data["box"][3] / camera_config.detect.height,
|
|
||||||
],
|
|
||||||
"label": event_data["label"],
|
|
||||||
"region": [
|
|
||||||
event_data["region"][0] / camera_config.detect.width,
|
|
||||||
event_data["region"][1] / camera_config.detect.height,
|
|
||||||
event_data["region"][2] / camera_config.detect.width,
|
|
||||||
event_data["region"][3] / camera_config.detect.height,
|
|
||||||
],
|
|
||||||
"zones": event_data["current_zones"],
|
|
||||||
},
|
|
||||||
).execute()
|
|
||||||
elif event_type == "end":
|
elif event_type == "end":
|
||||||
Timeline.insert(
|
timeline_entry[Timeline.class_type] = "gone"
|
||||||
timestamp=event_data["frame_time"],
|
Timeline.insert(timeline_entry).execute()
|
||||||
camera=camera,
|
|
||||||
source="tracked_object",
|
|
||||||
source_id=event_data["id"],
|
|
||||||
class_type="gone",
|
|
||||||
data={
|
|
||||||
"box": [
|
|
||||||
event_data["box"][0] / camera_config.detect.width,
|
|
||||||
event_data["box"][1] / camera_config.detect.height,
|
|
||||||
event_data["box"][2] / camera_config.detect.width,
|
|
||||||
event_data["box"][3] / camera_config.detect.height,
|
|
||||||
],
|
|
||||||
"label": event_data["label"],
|
|
||||||
"region": [
|
|
||||||
event_data["region"][0] / camera_config.detect.width,
|
|
||||||
event_data["region"][1] / camera_config.detect.height,
|
|
||||||
event_data["region"][2] / camera_config.detect.width,
|
|
||||||
event_data["region"][3] / camera_config.detect.height,
|
|
||||||
],
|
|
||||||
},
|
|
||||||
).execute()
|
|
||||||
|
@ -1065,3 +1065,14 @@ def get_tz_modifiers(tz_name: str) -> Tuple[str, str]:
|
|||||||
hour_modifier = f"{hours_offset} hour"
|
hour_modifier = f"{hours_offset} hour"
|
||||||
minute_modifier = f"{minutes_offset} minute"
|
minute_modifier = f"{minutes_offset} minute"
|
||||||
return hour_modifier, minute_modifier
|
return hour_modifier, minute_modifier
|
||||||
|
|
||||||
|
|
||||||
|
def to_relative_box(
|
||||||
|
width: int, height: int, box: Tuple[int, int, int, int]
|
||||||
|
) -> Tuple[int, int, int, int]:
|
||||||
|
return (
|
||||||
|
box[0] / width, # x
|
||||||
|
box[1] / height, # y
|
||||||
|
(box[2] - box[0]) / width, # w
|
||||||
|
(box[3] - box[1]) / height, # h
|
||||||
|
)
|
||||||
|
52
migrations/014_event_updates_for_fp.py
Normal file
52
migrations/014_event_updates_for_fp.py
Normal file
@ -0,0 +1,52 @@
|
|||||||
|
"""Peewee migrations
|
||||||
|
|
||||||
|
Some examples (model - class or model name)::
|
||||||
|
|
||||||
|
> Model = migrator.orm['model_name'] # Return model in current state by name
|
||||||
|
|
||||||
|
> migrator.sql(sql) # Run custom SQL
|
||||||
|
> migrator.python(func, *args, **kwargs) # Run python code
|
||||||
|
> migrator.create_model(Model) # Create a model (could be used as decorator)
|
||||||
|
> migrator.remove_model(model, cascade=True) # Remove a model
|
||||||
|
> migrator.add_fields(model, **fields) # Add fields to a model
|
||||||
|
> migrator.change_fields(model, **fields) # Change fields
|
||||||
|
> migrator.remove_fields(model, *field_names, cascade=True)
|
||||||
|
> migrator.rename_field(model, old_field_name, new_field_name)
|
||||||
|
> migrator.rename_table(model, new_table_name)
|
||||||
|
> migrator.add_index(model, *col_names, unique=False)
|
||||||
|
> migrator.drop_index(model, *col_names)
|
||||||
|
> migrator.add_not_null(model, *field_names)
|
||||||
|
> migrator.drop_not_null(model, *field_names)
|
||||||
|
> migrator.add_default(model, field_name, default)
|
||||||
|
|
||||||
|
"""
|
||||||
|
|
||||||
|
import datetime as dt
|
||||||
|
import peewee as pw
|
||||||
|
from playhouse.sqlite_ext import *
|
||||||
|
from decimal import ROUND_HALF_EVEN
|
||||||
|
from frigate.models import Event
|
||||||
|
|
||||||
|
try:
|
||||||
|
import playhouse.postgres_ext as pw_pext
|
||||||
|
except ImportError:
|
||||||
|
pass
|
||||||
|
|
||||||
|
SQL = pw.SQL
|
||||||
|
|
||||||
|
|
||||||
|
def migrate(migrator, database, fake=False, **kwargs):
|
||||||
|
migrator.add_fields(
|
||||||
|
Event,
|
||||||
|
score=pw.FloatField(null=True),
|
||||||
|
model_hash=pw.CharField(max_length=32, null=True),
|
||||||
|
detector_type=pw.CharField(max_length=32, null=True),
|
||||||
|
model_type=pw.CharField(max_length=32, null=True),
|
||||||
|
)
|
||||||
|
|
||||||
|
migrator.drop_not_null(Event, "area", "false_positive")
|
||||||
|
migrator.add_default(Event, "false_positive", 0)
|
||||||
|
|
||||||
|
|
||||||
|
def rollback(migrator, database, fake=False, **kwargs):
|
||||||
|
pass
|
@ -3,6 +3,7 @@ import { route } from 'preact-router';
|
|||||||
import ActivityIndicator from '../components/ActivityIndicator';
|
import ActivityIndicator from '../components/ActivityIndicator';
|
||||||
import Heading from '../components/Heading';
|
import Heading from '../components/Heading';
|
||||||
import { Tabs, TextTab } from '../components/Tabs';
|
import { Tabs, TextTab } from '../components/Tabs';
|
||||||
|
import Link from '../components/Link';
|
||||||
import { useApiHost } from '../api';
|
import { useApiHost } from '../api';
|
||||||
import useSWR from 'swr';
|
import useSWR from 'swr';
|
||||||
import useSWRInfinite from 'swr/infinite';
|
import useSWRInfinite from 'swr/infinite';
|
||||||
@ -57,7 +58,12 @@ export default function Events({ path, ...props }) {
|
|||||||
showDownloadMenu: false,
|
showDownloadMenu: false,
|
||||||
showDatePicker: false,
|
showDatePicker: false,
|
||||||
showCalendar: false,
|
showCalendar: false,
|
||||||
showPlusConfig: false,
|
showPlusSubmit: false,
|
||||||
|
});
|
||||||
|
const [plusSubmitEvent, setPlusSubmitEvent] = useState({
|
||||||
|
id: null,
|
||||||
|
label: null,
|
||||||
|
validBox: null,
|
||||||
});
|
});
|
||||||
const [uploading, setUploading] = useState([]);
|
const [uploading, setUploading] = useState([]);
|
||||||
const [viewEvent, setViewEvent] = useState();
|
const [viewEvent, setViewEvent] = useState();
|
||||||
@ -65,6 +71,8 @@ export default function Events({ path, ...props }) {
|
|||||||
const [eventDetailType, setEventDetailType] = useState('clip');
|
const [eventDetailType, setEventDetailType] = useState('clip');
|
||||||
const [downloadEvent, setDownloadEvent] = useState({
|
const [downloadEvent, setDownloadEvent] = useState({
|
||||||
id: null,
|
id: null,
|
||||||
|
label: null,
|
||||||
|
box: null,
|
||||||
has_clip: false,
|
has_clip: false,
|
||||||
has_snapshot: false,
|
has_snapshot: false,
|
||||||
plus_id: undefined,
|
plus_id: undefined,
|
||||||
@ -198,6 +206,8 @@ export default function Events({ path, ...props }) {
|
|||||||
e.stopPropagation();
|
e.stopPropagation();
|
||||||
setDownloadEvent((_prev) => ({
|
setDownloadEvent((_prev) => ({
|
||||||
id: event.id,
|
id: event.id,
|
||||||
|
box: event.box,
|
||||||
|
label: event.label,
|
||||||
has_clip: event.has_clip,
|
has_clip: event.has_clip,
|
||||||
has_snapshot: event.has_snapshot,
|
has_snapshot: event.has_snapshot,
|
||||||
plus_id: event.plus_id,
|
plus_id: event.plus_id,
|
||||||
@ -207,6 +217,16 @@ export default function Events({ path, ...props }) {
|
|||||||
setState({ ...state, showDownloadMenu: true });
|
setState({ ...state, showDownloadMenu: true });
|
||||||
};
|
};
|
||||||
|
|
||||||
|
const showSubmitToPlus = (event_id, label, box, e) => {
|
||||||
|
if (e) {
|
||||||
|
e.stopPropagation();
|
||||||
|
}
|
||||||
|
// if any of the box coordinates are > 1, then the box data is from an older version
|
||||||
|
// and not valid to submit to plus with the snapshot image
|
||||||
|
setPlusSubmitEvent({ id: event_id, label, validBox: !box.some((d) => d > 1) });
|
||||||
|
setState({ ...state, showDownloadMenu: false, showPlusSubmit: true });
|
||||||
|
};
|
||||||
|
|
||||||
const handleSelectDateRange = useCallback(
|
const handleSelectDateRange = useCallback(
|
||||||
(dates) => {
|
(dates) => {
|
||||||
setSearchParams({ ...searchParams, before: dates.before, after: dates.after });
|
setSearchParams({ ...searchParams, before: dates.before, after: dates.after });
|
||||||
@ -251,23 +271,16 @@ export default function Events({ path, ...props }) {
|
|||||||
[size, setSize, isValidating, isDone]
|
[size, setSize, isValidating, isDone]
|
||||||
);
|
);
|
||||||
|
|
||||||
const onSendToPlus = async (id, e) => {
|
const onSendToPlus = async (id, false_positive, validBox) => {
|
||||||
if (e) {
|
|
||||||
e.stopPropagation();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (uploading.includes(id)) {
|
if (uploading.includes(id)) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!config.plus.enabled) {
|
|
||||||
setState({ ...state, showDownloadMenu: false, showPlusConfig: true });
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
setUploading((prev) => [...prev, id]);
|
setUploading((prev) => [...prev, id]);
|
||||||
|
|
||||||
const response = await axios.post(`events/${id}/plus`);
|
const response = false_positive
|
||||||
|
? await axios.put(`events/${id}/false_positive`)
|
||||||
|
: await axios.post(`events/${id}/plus`, validBox ? { include_annotation: 1 } : {});
|
||||||
|
|
||||||
if (response.status === 200) {
|
if (response.status === 200) {
|
||||||
mutate(
|
mutate(
|
||||||
@ -289,6 +302,8 @@ export default function Events({ path, ...props }) {
|
|||||||
if (state.showDownloadMenu && downloadEvent.id === id) {
|
if (state.showDownloadMenu && downloadEvent.id === id) {
|
||||||
setState({ ...state, showDownloadMenu: false });
|
setState({ ...state, showDownloadMenu: false });
|
||||||
}
|
}
|
||||||
|
|
||||||
|
setState({ ...state, showPlusSubmit: false });
|
||||||
};
|
};
|
||||||
|
|
||||||
const handleEventDetailTabChange = (index) => {
|
const handleEventDetailTabChange = (index) => {
|
||||||
@ -375,12 +390,12 @@ export default function Events({ path, ...props }) {
|
|||||||
download
|
download
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
{(downloadEvent.end_time && downloadEvent.has_snapshot && !downloadEvent.plus_id) && (
|
{downloadEvent.end_time && downloadEvent.has_snapshot && !downloadEvent.plus_id && (
|
||||||
<MenuItem
|
<MenuItem
|
||||||
icon={UploadPlus}
|
icon={UploadPlus}
|
||||||
label={uploading.includes(downloadEvent.id) ? 'Uploading...' : 'Send to Frigate+'}
|
label={uploading.includes(downloadEvent.id) ? 'Uploading...' : 'Send to Frigate+'}
|
||||||
value="plus"
|
value="plus"
|
||||||
onSelect={() => onSendToPlus(downloadEvent.id)}
|
onSelect={() => showSubmitToPlus(downloadEvent.id, downloadEvent.label, downloadEvent.box)}
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
{downloadEvent.plus_id && (
|
{downloadEvent.plus_id && (
|
||||||
@ -435,8 +450,77 @@ export default function Events({ path, ...props }) {
|
|||||||
/>
|
/>
|
||||||
</Menu>
|
</Menu>
|
||||||
)}
|
)}
|
||||||
{state.showPlusConfig && (
|
{state.showPlusSubmit && (
|
||||||
<Dialog>
|
<Dialog>
|
||||||
|
{config.plus.enabled ? (
|
||||||
|
<>
|
||||||
|
<div className="p-4">
|
||||||
|
<Heading size="lg">Submit to Frigate+</Heading>
|
||||||
|
|
||||||
|
<img
|
||||||
|
className="flex-grow-0"
|
||||||
|
src={`${apiHost}/api/events/${plusSubmitEvent.id}/snapshot.jpg`}
|
||||||
|
alt={`${plusSubmitEvent.label}`}
|
||||||
|
/>
|
||||||
|
|
||||||
|
{plusSubmitEvent.validBox ? (
|
||||||
|
<p className="mb-2">
|
||||||
|
Objects in locations you want to avoid are not false positives. Submitting them as false positives
|
||||||
|
will confuse the model.
|
||||||
|
</p>
|
||||||
|
) : (
|
||||||
|
<p className="mb-2">
|
||||||
|
Events prior to version 0.13 can only be submitted to Frigate+ without annotations.
|
||||||
|
</p>
|
||||||
|
)}
|
||||||
|
</div>
|
||||||
|
{plusSubmitEvent.validBox ? (
|
||||||
|
<div className="p-2 flex justify-start flex-row-reverse space-x-2">
|
||||||
|
<Button className="ml-2" onClick={() => setState({ ...state, showPlusSubmit: false })} type="text">
|
||||||
|
{uploading.includes(plusSubmitEvent.id) ? 'Close' : 'Cancel'}
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
className="ml-2"
|
||||||
|
color="red"
|
||||||
|
onClick={() => onSendToPlus(plusSubmitEvent.id, true, plusSubmitEvent.validBox)}
|
||||||
|
disabled={uploading.includes(plusSubmitEvent.id)}
|
||||||
|
type="text"
|
||||||
|
>
|
||||||
|
This is not a {plusSubmitEvent.label}
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
className="ml-2"
|
||||||
|
color="green"
|
||||||
|
onClick={() => onSendToPlus(plusSubmitEvent.id, false, plusSubmitEvent.validBox)}
|
||||||
|
disabled={uploading.includes(plusSubmitEvent.id)}
|
||||||
|
type="text"
|
||||||
|
>
|
||||||
|
This is a {plusSubmitEvent.label}
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
) : (
|
||||||
|
<div className="p-2 flex justify-start flex-row-reverse space-x-2">
|
||||||
|
<Button
|
||||||
|
className="ml-2"
|
||||||
|
onClick={() => setState({ ...state, showPlusSubmit: false })}
|
||||||
|
disabled={uploading.includes(plusSubmitEvent.id)}
|
||||||
|
type="text"
|
||||||
|
>
|
||||||
|
{uploading.includes(plusSubmitEvent.id) ? 'Close' : 'Cancel'}
|
||||||
|
</Button>
|
||||||
|
<Button
|
||||||
|
className="ml-2"
|
||||||
|
onClick={() => onSendToPlus(plusSubmitEvent.id, false, plusSubmitEvent.validBox)}
|
||||||
|
disabled={uploading.includes(plusSubmitEvent.id)}
|
||||||
|
type="text"
|
||||||
|
>
|
||||||
|
Submit to Frigate+
|
||||||
|
</Button>
|
||||||
|
</div>
|
||||||
|
)}
|
||||||
|
</>
|
||||||
|
) : (
|
||||||
|
<>
|
||||||
<div className="p-4">
|
<div className="p-4">
|
||||||
<Heading size="lg">Setup a Frigate+ Account</Heading>
|
<Heading size="lg">Setup a Frigate+ Account</Heading>
|
||||||
<p className="mb-2">In order to submit images to Frigate+, you first need to setup an account.</p>
|
<p className="mb-2">In order to submit images to Frigate+, you first need to setup an account.</p>
|
||||||
@ -450,10 +534,12 @@ export default function Events({ path, ...props }) {
|
|||||||
</a>
|
</a>
|
||||||
</div>
|
</div>
|
||||||
<div className="p-2 flex justify-start flex-row-reverse space-x-2">
|
<div className="p-2 flex justify-start flex-row-reverse space-x-2">
|
||||||
<Button className="ml-2" onClick={() => setState({ ...state, showPlusConfig: false })} type="text">
|
<Button className="ml-2" onClick={() => setState({ ...state, showPlusSubmit: false })} type="text">
|
||||||
Close
|
Close
|
||||||
</Button>
|
</Button>
|
||||||
</div>
|
</div>
|
||||||
|
</>
|
||||||
|
)}
|
||||||
</Dialog>
|
</Dialog>
|
||||||
)}
|
)}
|
||||||
{deleteFavoriteState.showDeleteFavorite && (
|
{deleteFavoriteState.showDeleteFavorite && (
|
||||||
@ -539,12 +625,20 @@ export default function Events({ path, ...props }) {
|
|||||||
{event.end_time && event.has_snapshot && (
|
{event.end_time && event.has_snapshot && (
|
||||||
<Fragment>
|
<Fragment>
|
||||||
{event.plus_id ? (
|
{event.plus_id ? (
|
||||||
<div className="uppercase text-xs">Sent to Frigate+</div>
|
<div className="uppercase text-xs underline">
|
||||||
|
<Link
|
||||||
|
href={`https://plus.frigate.video/dashboard/edit-image/?id=${event.plus_id}`}
|
||||||
|
target="_blank"
|
||||||
|
rel="nofollow"
|
||||||
|
>
|
||||||
|
Edit in Frigate+
|
||||||
|
</Link>
|
||||||
|
</div>
|
||||||
) : (
|
) : (
|
||||||
<Button
|
<Button
|
||||||
color="gray"
|
color="gray"
|
||||||
disabled={uploading.includes(event.id)}
|
disabled={uploading.includes(event.id)}
|
||||||
onClick={(e) => onSendToPlus(event.id, e)}
|
onClick={(e) => showSubmitToPlus(event.id, event.label, event.box, e)}
|
||||||
>
|
>
|
||||||
{uploading.includes(event.id) ? 'Uploading...' : 'Send to Frigate+'}
|
{uploading.includes(event.id) ? 'Uploading...' : 'Send to Frigate+'}
|
||||||
</Button>
|
</Button>
|
||||||
@ -617,8 +711,12 @@ export default function Events({ path, ...props }) {
|
|||||||
style={{
|
style={{
|
||||||
left: `${Math.round(eventOverlay.data.box[0] * 100)}%`,
|
left: `${Math.round(eventOverlay.data.box[0] * 100)}%`,
|
||||||
top: `${Math.round(eventOverlay.data.box[1] * 100)}%`,
|
top: `${Math.round(eventOverlay.data.box[1] * 100)}%`,
|
||||||
right: `${Math.round((1 - eventOverlay.data.box[2]) * 100)}%`,
|
right: `${Math.round(
|
||||||
bottom: `${Math.round((1 - eventOverlay.data.box[3]) * 100)}%`,
|
(1 - eventOverlay.data.box[2] - eventOverlay.data.box[0]) * 100
|
||||||
|
)}%`,
|
||||||
|
bottom: `${Math.round(
|
||||||
|
(1 - eventOverlay.data.box[3] - eventOverlay.data.box[1]) * 100
|
||||||
|
)}%`,
|
||||||
}}
|
}}
|
||||||
>
|
>
|
||||||
{eventOverlay.class_type == 'entered_zone' ? (
|
{eventOverlay.class_type == 'entered_zone' ? (
|
||||||
|
Loading…
Reference in New Issue
Block a user