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

Makefile updates and refactoring

Browse Source
This commit is contained in:
YS 2022-01-11 12:30:05 +03:00
parent fbbcad49e8
commit 1c9749b219
3 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
Makefile
View File

@ -53,7 +53,7 @@ aarch64_dev:
aarch64_all: aarch64_wheels aarch64_ffmpeg aarch64_frigate
l4t_assets_yolo4:
aarch64_l4t_assets:
mkdir -p $$(pwd)/.l4t_assets
cp ./converters/yolo4/plugin/* .l4t_assets/
cp ./converters/yolo4/model/yolov4-tiny-416.trt .l4t_assets/yolov4-tiny-416.trt
@ -61,11 +61,19 @@ l4t_assets_yolo4:
# cp ./converters/yolo4/model/yolov4-416.trt .l4t_assets/yolov4-416.trt
# cp ./converters/yolo4/model/yolov4-288.trt .l4t_assets/yolov4-288.trt
l4t_dev: # l4t_assets_yolo4
nvidia-docker build --tag frigate.l4t --build-arg NGINX_VERSION=1.0.2 --file docker/Dockerfile.l4t.base .
aarch64_l4t_wheels:
@docker build --tag frigate-wheels-l4t --file docker/Dockerfile.wheels.l4t .
# Run l4t wheels using nvidia runtime
@docker rm frigate.wheels.l4t || true
@docker run --name frigate.wheels.l4t -it --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video --privileged frigate-wheels-l4t
# Commit changes to the container
@CONTAINER_ID=`docker ps -n 1 --format "{{.ID}}"`
@docker commit $$CONTAINER_ID frigate.wheels.l4t frigate-wheels-l4t:latest
@docker rm frigate.wheels.l4t || true
l4t_dev_test:
nvidia-docker build --tag frigate.l4t.onnx --build-arg NGINX_VERSION=1.0.2 --file docker/Dockerfile.l4t.onnx ./onnx_test/
aarch64_l4t_frigate: # aarch64_l4t_wheels aarch64_l4t_assets
#docker build --tag frigate-base-l4t --build-arg BASE_IMAGE=timongentzsch/l4t-ubuntu20-opencv:latest --build-arg FFMPEG_ARCH=arm64 --build-arg ARCH=aarch64 --build-arg WHEELS_VERSION=1.0.3 --build-arg NGINX_VERSION=1.0.2 --file docker/Dockerfile.base .
nvidia-docker build --tag frigate.l4t --build-arg NGINX_VERSION=1.0.2 --file docker/Dockerfile.aarch64.l4t .
armv7_wheels:

7
frigate/config.py
View File

@ -596,11 +596,10 @@ class CameraConfig(FrigateBaseModel):
else:
for input in self.gstreamer.inputs:
caps = gst_discover(input.path, ["width", "height", "video codec"])
logger.error(">>> caps %s", caps)
gst_cmd = self._get_gstreamer_cmd(self.gstreamer, input, caps)
if gst_cmd is None:
continue
logger.error(">>> gst_cmd %s", gst_cmd)
logger.debug("gstreamer command[%s] %s", self.name, gst_cmd)
self._decoder_cmds.append({"roles": input.roles, "cmd": gst_cmd})
@ -628,7 +627,9 @@ class CameraConfig(FrigateBaseModel):
else base_config.decoder_pipeline
)
decoder_pipeline = [part for part in decoder_pipeline if part != ""]
builder = builder.with_decoder_pipeline(decoder_pipeline, codec = caps.get("video codec"))
builder = builder.with_decoder_pipeline(
decoder_pipeline, codec=caps.get("video codec")
)
source_format_pipeline = (
gstreamer_input.source_format_pipeline

1
frigate/detection/__init__.py
View File

@ -106,6 +106,7 @@ def run_detector(
avg_speed.value = (avg_speed.value * 9 + duration) / 10
del object_detector
logger.debug("Object detector process[%s] exit.", name)
class DetectionProcess: