parent
2839a55803
commit
56e7da7594
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -1,348 +0,0 @@
|
|||||||
{
|
|
||||||
"cells": [
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 5,
|
|
||||||
"id": "817586c7",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"name": "stdout",
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: train (files: 6471)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: val (files: 548)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: test (files: 1610)\n",
|
|
||||||
"\n",
|
|
||||||
"📌 클래스 통계 결과:\n",
|
|
||||||
" - Class 0: 147747 개\n",
|
|
||||||
" - Class 1: 187005 개\n",
|
|
||||||
" - Class 2: 32702 개\n",
|
|
||||||
" - Class 3: 16284 개\n",
|
|
||||||
" - Class 4: 9117 개\n",
|
|
||||||
" - Class 5: 40378 개\n",
|
|
||||||
"\n",
|
|
||||||
"총 클래스 종류: 6\n"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"import os\n",
|
|
||||||
"from collections import Counter\n",
|
|
||||||
"\n",
|
|
||||||
"# label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels copy\"\n",
|
|
||||||
"label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels\"\n",
|
|
||||||
"splits = [\"train\", \"val\",'test']\n",
|
|
||||||
"\n",
|
|
||||||
"class_counter = Counter()\n",
|
|
||||||
"\n",
|
|
||||||
"for split in splits:\n",
|
|
||||||
" split_path = os.path.join(label_root, split)\n",
|
|
||||||
" \n",
|
|
||||||
" # 라벨 txt 파일 탐색\n",
|
|
||||||
" label_files = [f for f in os.listdir(split_path) if f.endswith(\".txt\")]\n",
|
|
||||||
" \n",
|
|
||||||
" print(f\"\\n🔍 Checking split: {split} (files: {len(label_files)})\")\n",
|
|
||||||
"\n",
|
|
||||||
" for file in label_files:\n",
|
|
||||||
" file_path = os.path.join(split_path, file)\n",
|
|
||||||
" with open(file_path, \"r\") as f:\n",
|
|
||||||
" for line in f.readlines():\n",
|
|
||||||
" if line.strip(): # 빈 줄 제거\n",
|
|
||||||
" class_id = line.split()[0] # 첫번째 값 = 클래스\n",
|
|
||||||
" class_counter[class_id] += 1\n",
|
|
||||||
"\n",
|
|
||||||
"# 최종 결과 출력\n",
|
|
||||||
"print(\"\\n📌 클래스 통계 결과:\")\n",
|
|
||||||
"for cls, count in sorted(class_counter.items(), key=lambda x: int(x[0])):\n",
|
|
||||||
" print(f\" - Class {cls}: {count} 개\")\n",
|
|
||||||
"\n",
|
|
||||||
"print(f\"\\n총 클래스 종류: {len(class_counter)}\")\n"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 6,
|
|
||||||
"id": "35cf2381",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"name": "stdout",
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"\n",
|
|
||||||
"📂 Updating: train (files: 6471)\n",
|
|
||||||
"\n",
|
|
||||||
"📂 Updating: val (files: 548)\n",
|
|
||||||
"\n",
|
|
||||||
"📂 Updating: test (files: 1610)\n",
|
|
||||||
"\n",
|
|
||||||
"✅ 라벨 클래스 번호 재정렬 완료!\n"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"import os\n",
|
|
||||||
"\n",
|
|
||||||
"label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels\"\n",
|
|
||||||
"splits = [\"train\", \"val\", \"test\"]\n",
|
|
||||||
"\n",
|
|
||||||
"# 클래스 재매핑 설정\n",
|
|
||||||
"mapping = {\n",
|
|
||||||
" \"0\": \"0\", # person stays 0\n",
|
|
||||||
" \"1\": \"1\", # car stays 1\n",
|
|
||||||
" \"2\": \"5\", # van -> 5\n",
|
|
||||||
" \"3\": \"2\", # truck -> 2\n",
|
|
||||||
" \"4\": \"3\", # bus -> 3\n",
|
|
||||||
" \"5\": \"4\" # motor -> 4\n",
|
|
||||||
"}\n",
|
|
||||||
"\n",
|
|
||||||
"for split in splits:\n",
|
|
||||||
" split_path = os.path.join(label_root, split)\n",
|
|
||||||
" label_files = [f for f in os.listdir(split_path) if f.endswith(\".txt\")]\n",
|
|
||||||
"\n",
|
|
||||||
" print(f\"\\n📂 Updating: {split} (files: {len(label_files)})\")\n",
|
|
||||||
"\n",
|
|
||||||
" for file in label_files:\n",
|
|
||||||
" file_path = os.path.join(split_path, file)\n",
|
|
||||||
"\n",
|
|
||||||
" new_lines = []\n",
|
|
||||||
" with open(file_path, \"r\") as f:\n",
|
|
||||||
" for line in f.readlines():\n",
|
|
||||||
" if line.strip():\n",
|
|
||||||
" parts = line.split()\n",
|
|
||||||
" cls = parts[0]\n",
|
|
||||||
" parts[0] = mapping[cls] # 클래스 번호 변경\n",
|
|
||||||
" new_lines.append(\" \".join(parts) + \"\\n\")\n",
|
|
||||||
"\n",
|
|
||||||
" # 덮어쓰기\n",
|
|
||||||
" with open(file_path, \"w\") as f:\n",
|
|
||||||
" f.writelines(new_lines)\n",
|
|
||||||
"\n",
|
|
||||||
"print(\"\\n✅ 라벨 클래스 번호 재정렬 완료!\")\n"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 7,
|
|
||||||
"id": "3fdd6e44",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"name": "stdout",
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: train (files: 6471)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: val (files: 548)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: test (files: 1610)\n",
|
|
||||||
"\n",
|
|
||||||
"📌 클래스 통계 결과:\n",
|
|
||||||
" - Class 0: 147747 개\n",
|
|
||||||
" - Class 1: 187005 개\n",
|
|
||||||
" - Class 2: 16284 개\n",
|
|
||||||
" - Class 3: 9117 개\n",
|
|
||||||
" - Class 4: 40378 개\n",
|
|
||||||
" - Class 5: 32702 개\n",
|
|
||||||
"\n",
|
|
||||||
"총 클래스 종류: 6\n"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"import os\n",
|
|
||||||
"from collections import Counter\n",
|
|
||||||
"\n",
|
|
||||||
"# label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels copy\"\n",
|
|
||||||
"label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels\"\n",
|
|
||||||
"splits = [\"train\", \"val\",'test']\n",
|
|
||||||
"\n",
|
|
||||||
"class_counter = Counter()\n",
|
|
||||||
"\n",
|
|
||||||
"for split in splits:\n",
|
|
||||||
" split_path = os.path.join(label_root, split)\n",
|
|
||||||
" \n",
|
|
||||||
" # 라벨 txt 파일 탐색\n",
|
|
||||||
" label_files = [f for f in os.listdir(split_path) if f.endswith(\".txt\")]\n",
|
|
||||||
" \n",
|
|
||||||
" print(f\"\\n🔍 Checking split: {split} (files: {len(label_files)})\")\n",
|
|
||||||
"\n",
|
|
||||||
" for file in label_files:\n",
|
|
||||||
" file_path = os.path.join(split_path, file)\n",
|
|
||||||
" with open(file_path, \"r\") as f:\n",
|
|
||||||
" for line in f.readlines():\n",
|
|
||||||
" if line.strip(): # 빈 줄 제거\n",
|
|
||||||
" class_id = line.split()[0] # 첫번째 값 = 클래스\n",
|
|
||||||
" class_counter[class_id] += 1\n",
|
|
||||||
"\n",
|
|
||||||
"# 최종 결과 출력\n",
|
|
||||||
"print(\"\\n📌 클래스 통계 결과:\")\n",
|
|
||||||
"for cls, count in sorted(class_counter.items(), key=lambda x: int(x[0])):\n",
|
|
||||||
" print(f\" - Class {cls}: {count} 개\")\n",
|
|
||||||
"\n",
|
|
||||||
"print(f\"\\n총 클래스 종류: {len(class_counter)}\")\n"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 1,
|
|
||||||
"id": "ca796e3d",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"name": "stdout",
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"Processing train: 6471 files\n",
|
|
||||||
"Processing val: 548 files\n",
|
|
||||||
"Processing test: 1610 files\n",
|
|
||||||
"Done. All label files updated!\n"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"import os\n",
|
|
||||||
"from glob import glob\n",
|
|
||||||
"\n",
|
|
||||||
"# 라벨 매핑\n",
|
|
||||||
"mapping = {\n",
|
|
||||||
" 0: 0,\n",
|
|
||||||
" 1: 2,\n",
|
|
||||||
" 2: 5,\n",
|
|
||||||
" 3: 4,\n",
|
|
||||||
" 4: 3,\n",
|
|
||||||
" 5: 1\n",
|
|
||||||
"}\n",
|
|
||||||
"\n",
|
|
||||||
"# labels 폴더 경로\n",
|
|
||||||
"base_dir = \"/home/cuuva/experiment/datasets/VisDrone/labels\"\n",
|
|
||||||
"\n",
|
|
||||||
"# train, val, test 모두 처리\n",
|
|
||||||
"splits = [\"train\", \"val\", \"test\"]\n",
|
|
||||||
"\n",
|
|
||||||
"for split in splits:\n",
|
|
||||||
" label_dir = os.path.join(base_dir, split)\n",
|
|
||||||
" txt_files = glob(os.path.join(label_dir, \"*.txt\"))\n",
|
|
||||||
"\n",
|
|
||||||
" print(f\"Processing {split}: {len(txt_files)} files\")\n",
|
|
||||||
"\n",
|
|
||||||
" for txt_path in txt_files:\n",
|
|
||||||
" lines = []\n",
|
|
||||||
" with open(txt_path, \"r\") as f:\n",
|
|
||||||
" for line in f.readlines():\n",
|
|
||||||
" parts = line.strip().split()\n",
|
|
||||||
" if len(parts) < 5:\n",
|
|
||||||
" continue\n",
|
|
||||||
"\n",
|
|
||||||
" old_cls = int(parts[0])\n",
|
|
||||||
" new_cls = mapping[old_cls]\n",
|
|
||||||
"\n",
|
|
||||||
" # 클래스만 변경해서 다시 저장\n",
|
|
||||||
" parts[0] = str(new_cls)\n",
|
|
||||||
" lines.append(\" \".join(parts))\n",
|
|
||||||
"\n",
|
|
||||||
" with open(txt_path, \"w\") as f:\n",
|
|
||||||
" f.write(\"\\n\".join(lines))\n",
|
|
||||||
"\n",
|
|
||||||
"print(\"Done. All label files updated!\")\n"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": 2,
|
|
||||||
"id": "a461857c",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [
|
|
||||||
{
|
|
||||||
"name": "stdout",
|
|
||||||
"output_type": "stream",
|
|
||||||
"text": [
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: train (files: 6471)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: val (files: 548)\n",
|
|
||||||
"\n",
|
|
||||||
"🔍 Checking split: test (files: 1610)\n",
|
|
||||||
"\n",
|
|
||||||
"📌 클래스 통계 결과:\n",
|
|
||||||
" - Class 0: 147747 개\n",
|
|
||||||
" - Class 1: 32702 개\n",
|
|
||||||
" - Class 2: 187005 개\n",
|
|
||||||
" - Class 3: 40378 개\n",
|
|
||||||
" - Class 4: 9117 개\n",
|
|
||||||
" - Class 5: 16284 개\n",
|
|
||||||
"\n",
|
|
||||||
"총 클래스 종류: 6\n"
|
|
||||||
]
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"source": [
|
|
||||||
"import os\n",
|
|
||||||
"from collections import Counter\n",
|
|
||||||
"\n",
|
|
||||||
"# label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels copy\"\n",
|
|
||||||
"label_root = \"/home/cuuva/experiment/datasets/VisDrone/labels\"\n",
|
|
||||||
"splits = [\"train\", \"val\",'test']\n",
|
|
||||||
"\n",
|
|
||||||
"class_counter = Counter()\n",
|
|
||||||
"\n",
|
|
||||||
"for split in splits:\n",
|
|
||||||
" split_path = os.path.join(label_root, split)\n",
|
|
||||||
" \n",
|
|
||||||
" # 라벨 txt 파일 탐색\n",
|
|
||||||
" label_files = [f for f in os.listdir(split_path) if f.endswith(\".txt\")]\n",
|
|
||||||
" \n",
|
|
||||||
" print(f\"\\n🔍 Checking split: {split} (files: {len(label_files)})\")\n",
|
|
||||||
"\n",
|
|
||||||
" for file in label_files:\n",
|
|
||||||
" file_path = os.path.join(split_path, file)\n",
|
|
||||||
" with open(file_path, \"r\") as f:\n",
|
|
||||||
" for line in f.readlines():\n",
|
|
||||||
" if line.strip(): # 빈 줄 제거\n",
|
|
||||||
" class_id = line.split()[0] # 첫번째 값 = 클래스\n",
|
|
||||||
" class_counter[class_id] += 1\n",
|
|
||||||
"\n",
|
|
||||||
"# 최종 결과 출력\n",
|
|
||||||
"print(\"\\n📌 클래스 통계 결과:\")\n",
|
|
||||||
"for cls, count in sorted(class_counter.items(), key=lambda x: int(x[0])):\n",
|
|
||||||
" print(f\" - Class {cls}: {count} 개\")\n",
|
|
||||||
"\n",
|
|
||||||
"print(f\"\\n총 클래스 종류: {len(class_counter)}\")\n"
|
|
||||||
]
|
|
||||||
},
|
|
||||||
{
|
|
||||||
"cell_type": "code",
|
|
||||||
"execution_count": null,
|
|
||||||
"id": "8320460f",
|
|
||||||
"metadata": {},
|
|
||||||
"outputs": [],
|
|
||||||
"source": []
|
|
||||||
}
|
|
||||||
],
|
|
||||||
"metadata": {
|
|
||||||
"kernelspec": {
|
|
||||||
"display_name": "1stagedetect",
|
|
||||||
"language": "python",
|
|
||||||
"name": "python3"
|
|
||||||
},
|
|
||||||
"language_info": {
|
|
||||||
"codemirror_mode": {
|
|
||||||
"name": "ipython",
|
|
||||||
"version": 3
|
|
||||||
},
|
|
||||||
"file_extension": ".py",
|
|
||||||
"mimetype": "text/x-python",
|
|
||||||
"name": "python",
|
|
||||||
"nbconvert_exporter": "python",
|
|
||||||
"pygments_lexer": "ipython3",
|
|
||||||
"version": "3.10.18"
|
|
||||||
}
|
|
||||||
},
|
|
||||||
"nbformat": 4,
|
|
||||||
"nbformat_minor": 5
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue