HQU-gxy/ChArUcoBoardExp
7
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

cute object we go

Browse Source
This commit is contained in:
crosstyan
2024-12-18 15:40:51 +08:00
parent bfac2c3b60
commit a207c90cb9
3 changed files with 177 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

169
boom.ipynb
View File

@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "code",
"execution_count": 142,
"execution_count": 1,
"metadata": {},
"outputs": [],
"source": [
@ -118,7 +118,7 @@
},
{
"cell_type": "code",
"execution_count": 143,
"execution_count": 2,
"metadata": {},
"outputs": [
{
@ -155,7 +155,7 @@
},
{
"cell_type": "code",
"execution_count": 144,
"execution_count": 3,
"metadata": {},
"outputs": [],
"source": [
@ -284,7 +284,7 @@
},
{
"cell_type": "code",
"execution_count": 145,
"execution_count": 4,
"metadata": {},
"outputs": [],
"source": [
@ -314,7 +314,7 @@
},
{
"cell_type": "code",
"execution_count": 146,
"execution_count": 15,
"metadata": {},
"outputs": [
{
@ -331,10 +331,10 @@
"mode": "markers+lines+text",
"name": "16",
"text": [
"0",
"1",
"2",
"3"
"16:0",
"16:1",
"16:2",
"16:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -364,10 +364,10 @@
"mode": "markers+lines+text",
"name": "17",
"text": [
"0",
"1",
"2",
"3"
"17:0",
"17:1",
"17:2",
"17:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -397,10 +397,10 @@
"mode": "markers+lines+text",
"name": "18",
"text": [
"0",
"1",
"2",
"3"
"18:0",
"18:1",
"18:2",
"18:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -430,10 +430,10 @@
"mode": "markers+lines+text",
"name": "19",
"text": [
"0",
"1",
"2",
"3"
"19:0",
"19:1",
"19:2",
"19:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -483,10 +483,10 @@
"mode": "markers+lines+text",
"name": "20",
"text": [
"0",
"1",
"2",
"3"
"20:0",
"20:1",
"20:2",
"20:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -516,10 +516,10 @@
"mode": "markers+lines+text",
"name": "21",
"text": [
"0",
"1",
"2",
"3"
"21:0",
"21:1",
"21:2",
"21:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -549,10 +549,10 @@
"mode": "markers+lines+text",
"name": "22",
"text": [
"0",
"1",
"2",
"3"
"22:0",
"22:1",
"22:2",
"22:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -582,10 +582,10 @@
"mode": "markers+lines+text",
"name": "23",
"text": [
"0",
"1",
"2",
"3"
"23:0",
"23:1",
"23:2",
"23:3"
],
"textposition": "middle center",
"type": "scatter3d",
@ -635,10 +635,10 @@
"mode": "markers+lines+text",
"name": "24",
"text": [
"0",
"1",
"2",
"3"
"24:0",
"24:1",
"24:2",
"24:3"
],
"type": "scatter3d",
"x": [
@ -667,10 +667,10 @@
"mode": "markers+lines+text",
"name": "25",
"text": [
"0",
"1",
"2",
"3"
"25:0",
"25:1",
"25:2",
"25:3"
],
"type": "scatter3d",
"x": [
@ -699,10 +699,10 @@
"mode": "markers+lines+text",
"name": "26",
"text": [
"0",
"1",
"2",
"3"
"26:0",
"26:1",
"26:2",
"26:3"
],
"type": "scatter3d",
"x": [
@ -731,10 +731,10 @@
"mode": "markers+lines+text",
"name": "27",
"text": [
"0",
"1",
"2",
"3"
"27:0",
"27:1",
"27:2",
"27:3"
],
"type": "scatter3d",
"x": [
@ -1625,15 +1625,15 @@
],
"source": [
"fig = go.Figure()\n",
"t_corners = plane_a.transformed_corners\n",
"for i, corners in enumerate(t_corners):\n",
"t_corners_a = plane_a.transformed_corners\n",
"for i, corners in enumerate(t_corners_a):\n",
" fig.add_trace(\n",
" go.Scatter3d(\n",
" x=corners[:, 0],\n",
" y=corners[:, 1],\n",
" z=corners[:, 2],\n",
" mode=\"markers+lines+text\",\n",
" text=list(range(4)),\n",
" text=list(map(lambda x: f\"{plane_a.ids[i]}:{x}\", range(4))),\n",
" textposition=\"middle center\",\n",
" name=str(plane_a.ids[i]),\n",
" marker=dict(size=1),\n",
@ -1660,7 +1660,7 @@
" y=corners[:, 1],\n",
" z=corners[:, 2],\n",
" mode=\"markers+lines+text\",\n",
" text=list(range(4)),\n",
" text=list(map(lambda x: f\"{plane_b.ids[i]}:{x}\", range(4))),\n",
" textposition=\"middle center\",\n",
" name=str(plane_b.ids[i]),\n",
" marker=dict(size=1),\n",
@ -1685,7 +1685,7 @@
" y=corners[:, 1],\n",
" z=corners[:, 2],\n",
" mode=\"markers+lines+text\",\n",
" text=list(range(4)),\n",
" text=list(map(lambda x: f\"{plane_c.ids[i]}:{x}\", range(4))),\n",
" name=str(plane_c.ids[i]),\n",
" marker=dict(size=1),\n",
" )\n",
@ -1719,6 +1719,59 @@
")\n",
"fig.show()"
]
},
{
"cell_type": "code",
"execution_count": 16,
"metadata": {},
"outputs": [
{
"data": {
"text/html": [
"<pre>[{name: &#x27;a&#x27;, ids: [16, 17, 18, 19], corners: [[...], ...]},\n",
" {name: &#x27;b&#x27;, ids: [20, 21, 22, 23], corners: [[...], ...]},\n",
" {name: &#x27;c&#x27;, ids: [24, 25, 26, 27], corners: [[...], ...]}]\n",
"-----------------------------------------------------------\n",
"type: 3 * {\n",
" name: string,\n",
" ids: var * int64,\n",
" corners: var * var * var * float64\n",
"}</pre>"
],
"text/plain": [
"<Array [{name: 'a', ids: [...], ...}, ...] type='3 * {name: string, ids: va...'>"
]
},
"metadata": {},
"output_type": "display_data"
}
],
"source": [
"import awkward as ak\n",
"from awkward import Record as AwkwardRecord, Array as AwkwardArray\n",
"\n",
"coords = AwkwardArray(\n",
" [\n",
" {\n",
" \"name\": \"a\",\n",
" \"ids\": plane_a.ids,\n",
" \"corners\": t_corners_a,\n",
" },\n",
" {\n",
" \"name\": \"b\",\n",
" \"ids\": plane_b.ids,\n",
" \"corners\": t_corners_b,\n",
" },\n",
" {\n",
" \"name\": \"c\",\n",
" \"ids\": plane_c.ids,\n",
" \"corners\": t_corners_c,\n",
" },\n",
" ]\n",
")\n",
"display(coords)\n",
"_ = ak.to_parquet(coords, \"output/object_points.parquet\")"
]
}
],
"metadata": {