{ "cells": [ { "cell_type": "code", "execution_count": 34, "metadata": {}, "outputs": [], "source": [ "from datetime import datetime, timedelta\n", "from pathlib import Path\n", "from typing import Generator, Sequence, TypeAlias, TypedDict\n", "\n", "import awkward as ak\n", "import jax\n", "import jax.numpy as jnp\n", "import numpy as np\n", "from jaxtyping import Array, Num\n", "from matplotlib import pyplot as plt\n", "\n", "from app.camera import Detection\n", "from app.camera import Camera, CameraParams\n", "\n", "NDArray: TypeAlias = np.ndarray" ] }, { "cell_type": "code", "execution_count": 35, "metadata": {}, "outputs": [ { "data": { "text/html": [ "
[{name: 'AE_01', port: 5602, intrinsic: {...}, extrinsic: {...}, ...},\n",
" {name: 'AE_1A', port: 5601, intrinsic: {...}, extrinsic: {...}, ...},\n",
" {name: 'AE_08', port: 5600, intrinsic: {...}, extrinsic: {...}, ...}]\n",
"------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------\n",
"backend: cpu\n",
"nbytes: 823 B\n",
"type: 3 * {\n",
" name: string,\n",
" port: int64,\n",
" intrinsic: {\n",
" camera_matrix: var * var * var * float64,\n",
" distortion_coefficients: var * float64\n",
" },\n",
" extrinsic: {\n",
" rvec: var * float64,\n",
" tvec: var * float64\n",
" },\n",
" resolution: {\n",
" width: int64,\n",
" height: int64\n",
" }\n",
"}"
],
"text/plain": [
"[{frame_index: 0, boxes: [[599, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 1, boxes: [[599, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 2, boxes: [[599, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 3, boxes: [[599, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 4, boxes: [[598, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 5, boxes: [[596, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 6, boxes: [[594, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 7, boxes: [[595, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 8, boxes: [[595, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" {frame_index: 9, boxes: [[595, ...], [...]], kps: [...], kps_scores: ..., ...},\n",
" ...,\n",
" {frame_index: 520, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 521, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 522, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 523, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 524, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 525, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 526, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 527, boxes: [[1.09e+03, ...], ...], kps: [...], ...},\n",
" {frame_index: 528, boxes: [[1.09e+03, ...], ...], kps: [...], ...}]\n",
"-----------------------------------------------------------------------------------------------------------------------------------------------\n",
"backend: cpu\n",
"nbytes: 4.6 MB\n",
"type: 529 * {\n",
" frame_index: int64,\n",
" boxes: var * var * float64,\n",
" kps: var * var * var * float64,\n",
" kps_scores: var * var * float64\n",
"}"
],
"text/plain": [
"