sk/place6D_Nurburgring
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

dataset_88kps+some-updates

Browse Source
This commit is contained in:
Кирилл Шохин
2022-08-14 09:59:01 +03:00
parent ac04526fe1
commit 998aec2b80
943 changed files with 945 additions and 938 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
Karussell/Supervisely/Bbox from Pose.ipynb
View File

@@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "code",
"execution_count": 5,
"execution_count": 1,
"id": "07bdc0da",
"metadata": {},
"outputs": [],
@@ -11,17 +11,17 @@
"import json\n",
"import numpy as np\n",
"from tqdm.notebook import tqdm\n",
"path_dataset = r'D:\\downloads\\Nurburg-karussel'\n",
"path_dataset = r'D:\\Karusel\\Nurburg-karussel_88'\n",
"from SuperviselyKeypointsGUI.SuperviselyKeypointsGUI import *\n",
"\n",
"\n",
"keypoints_3d_path = r'SuperviselyKeypointsGUI\\karussel_24kps.csv'\n",
"keypoints_3d_path = r'C:\\Users\\Kir\\Jupiter\\Nurburg\\keypoints_88\\karussel_88kps.csv'\n",
"keypoints_3d = pd.read_csv(keypoints_3d_path, index_col=0).astype(float)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"execution_count": 2,
"id": "ca890aa0",
"metadata": {},
"outputs": [],
@@ -34,7 +34,7 @@
},
{
"cell_type": "code",
"execution_count": 9,
"execution_count": 3,
"id": "b8135031",
"metadata": {},
"outputs": [],
@@ -60,7 +60,7 @@
},
{
"cell_type": "code",
"execution_count": 8,
"execution_count": 4,
"id": "79762c08",
"metadata": {},
"outputs": [],
@@ -79,14 +79,14 @@
},
{
"cell_type": "code",
"execution_count": 10,
"execution_count": 5,
"id": "f9440b54",
"metadata": {},
"outputs": [
{
"data": {
"application/vnd.jupyter.widget-view+json": {
"model_id": "36c51c895f4e4f6b806c2af9e7bc14b9",
"model_id": "1a27ecccf8054a5082d0a51178c0f2a9",
"version_major": 2,
"version_minor": 0
},
@@ -111,13 +111,13 @@
},
{
"cell_type": "code",
"execution_count": 11,
"execution_count": 7,
"id": "dfcb675b",
"metadata": {},
"outputs": [],
"source": [
"# with open(os.path.join(path_dataset, 'karusel_COCO.json'), 'w') as file:\n",
"# json.dump(coco, file)"
"with open(os.path.join(path_dataset, 'karusel_COCO.json'), 'w') as file:\n",
" json.dump(coco, file)"
]
},
{
@@ -130,7 +130,7 @@
},
{
"cell_type": "code",
"execution_count": 12,
"execution_count": 8,
"id": "3aec7a2b",
"metadata": {},
"outputs": [],
@@ -165,7 +165,7 @@
},
{
"cell_type": "code",
"execution_count": 23,
"execution_count": 9,
"id": "aafd54fb",
"metadata": {},
"outputs": [],
@@ -175,7 +175,7 @@
},
{
"cell_type": "code",
"execution_count": 24,
"execution_count": 10,
"id": "f6af98fc",
"metadata": {},
"outputs": [],
@@ -184,7 +184,7 @@
"train_path_images = [os.path.join(path_dataset, 'images', 'img', x['file_name']) for x in train['images']]\n",
"\n",
"import shutil\n",
"path_new_dataset = r'C:\\Users\\Kir\\Jupiter\\Drivecast\\place6D_Nurburgring\\Karussell\\Training\\Karusel_dataset_v2'\n",
"path_new_dataset = r'C:\\Users\\Kir\\Jupiter\\Drivecast\\place6D_Nurbsurgring\\Karussell\\Training\\Karusel_dataset_v2_88'\n",
"\n",
"path_train_img = os.path.join(path_new_dataset, 'images', 'train')\n",
"path_test_img = os.path.join(path_new_dataset, 'images', 'val')\n",