Person Counter using torch
Project description
Person counter
from opencv_stream import VideoStreamer, FpsDrawer from person_counter.model import PersonCounterModel, PersonCounterOutput import numpy as np import os
VIDEO_DIR = "D:/project/facebodydetection/facebodydetect/app/src/videos" def get_video(): paths = [ os.path.join(VIDEO_DIR, p) for p in os.listdir(VIDEO_DIR)] return np.random.choice(paths)
stream = VideoStreamer.from_video_input(get_video()) fps = FpsDrawer()
model = PersonCounterModel()
@stream.on_next_frame() def index(frame: np.ndarray):
result = model.predict(frame)
if result.is_ok(): output: PersonCounterOutput = result.unwrap() output.draw(frame) else: raise result.exception
fps.draw(frame)
stream.start()
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for person_counter-1.1.19-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | eda214ab319a8eb6e33f091e65e15241ac60e9e9496e3f99cf2065b71b1c5876 |
|
MD5 | 330d9c0d70a5b670b0b0a8b39feca4f7 |
|
BLAKE2b-256 | e9f8fc511b1c09074193d2f5e8e249a74d710f11f2b972e8b9e750a2e57ae7a1 |