feat: change strateifiedkfold save path

#47

tools/stratifiedgroupkfold.ipynb

@@ -15,15 +15,20 @@
  },
  {
  "cell_type": "code",
- "execution_count": 3,
+ "execution_count": 2,
  "metadata": {},
  "outputs": [],
  "source": [
+ "# annotation file\n",
+ "annotation = '/data/ephemeral/home/relabeled_general_trash_json/train_relabeled_class_20.json'\n",
+ "\n",
  "# data annotation file 저장 경로\n",
- "save_annotation_path = '/data/ephemeral/home/dataset'\n",
+ "save_folder = annotation.split(\"/\")[-1].split(\".\")[0]\n",
+ "save_annotation_path = f'/data/ephemeral/home/dataset/{save_folder}_k_fold'\n",
  "\n",
- "# annotation file\n",
- "annotation = '/data/ephemeral/home/dataset/train.json'\n",
+ "# 저장 경로에 폴더가 없으면 생성\n",
+ "if not os.path.exists(save_annotation_path):\n",
+ " os.makedirs(save_annotation_path)\n",
  "\n",
  "with open(annotation) as f:\n",
  " data = json.load(f)\n",
@@ -36,7 +41,7 @@
  },
  {
  "cell_type": "code",
- "execution_count": 4,
+ "execution_count": 3,
  "metadata": {},
  "outputs": [],
  "source": [
@@ -57,7 +62,7 @@
  },
  {
  "cell_type": "code",
- "execution_count": 6,
+ "execution_count": 4,
  "metadata": {},
  "outputs": [
  {