chris/MiniGPT-4
1
0
Fork 0
mirror of https://github.com/Vision-CAIR/MiniGPT-4.git synced 2025-04-04 18:10:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

add normalize bbox

Browse Source
This commit is contained in:
ThuanNaN 2025-01-13 11:07:02 +07:00
parent 3718f9ff69
commit 8415c675a5
1 changed files with 37 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
create_dataset.ipynb
View File

@ -2,7 +2,7 @@
"cells": [
{
"cell_type": "code",
"execution_count": 48,
"execution_count": 6,
"metadata": {},
"outputs": [],
"source": [
@ -15,7 +15,7 @@
},
{
"cell_type": "code",
"execution_count": 49,
"execution_count": 7,
"metadata": {},
"outputs": [],
"source": [
@ -30,7 +30,25 @@
},
{
"cell_type": "code",
"execution_count": 50,
"execution_count": 8,
"metadata": {},
"outputs": [],
"source": [
"def normalize_bbox(bbox, width, height):\n",
" \"\"\"\n",
" Normalize bounding box to the range [0, 100].\n",
" \"\"\"\n",
" return [\n",
" int((bbox[0] / width) * 100), # x_min\n",
" int((bbox[1] / height) * 100), # y_min\n",
" int((bbox[2] / width) * 100), # x_max\n",
" int((bbox[3] / height) * 100), # y_max\n",
" ]"
]
},
{
"cell_type": "code",
"execution_count": 9,
"metadata": {},
"outputs": [
{
@ -144,7 +162,7 @@
},
{
"cell_type": "code",
"execution_count": 51,
"execution_count": 10,
"metadata": {},
"outputs": [
{
@ -153,7 +171,7 @@
"5354"
]
},
"execution_count": 51,
"execution_count": 10,
"metadata": {},
"output_type": "execute_result"
}
@ -164,7 +182,7 @@
},
{
"cell_type": "code",
"execution_count": 52,
"execution_count": 11,
"metadata": {},
"outputs": [],
"source": [
@ -178,17 +196,22 @@
},
{
"cell_type": "code",
"execution_count": 53,
"execution_count": 12,
"metadata": {},
"outputs": [],
"source": [
"for data in mvtech_ad_data_for_regression:\n",
" data[\"image_path\"] = data[\"image_path\"].replace(\"MVTEC_det\", \".\")"
" data[\"image_path\"] = data[\"image_path\"].replace(\"MVTEC_det\", \".\")\n",
" data[\"bbox\"] = normalize_bbox(\n",
" bbox=data[\"bbox\"],\n",
" width=data[\"width\"],\n",
" height=data[\"height\"]\n",
" )"
]
},
{
"cell_type": "code",
"execution_count": 54,
"execution_count": 13,
"metadata": {},
"outputs": [],
"source": [
@ -198,7 +221,7 @@
},
{
"cell_type": "code",
"execution_count": 55,
"execution_count": 14,
"metadata": {},
"outputs": [],
"source": [
@ -213,7 +236,7 @@
},
{
"cell_type": "code",
"execution_count": 56,
"execution_count": 15,
"metadata": {},
"outputs": [],
"source": [
@ -240,7 +263,7 @@
},
{
"cell_type": "code",
"execution_count": 57,
"execution_count": 16,
"metadata": {},
"outputs": [],
"source": [
@ -279,7 +302,7 @@
},
{
"cell_type": "code",
"execution_count": 58,
"execution_count": 17,
"metadata": {},
"outputs": [
{
@ -288,7 +311,7 @@
"Text(0.5, 0, 'Good/Defect')"
]
},
"execution_count": 58,
"execution_count": 17,
"metadata": {},
"output_type": "execute_result"
},