Forráskód Böngészése

refactor(cuhkpedes): 更新 CUHK-PEDES 数据集处理笔记本

- 在处理数据集时添加了打印训练、测试和验证集图像数量的代码
- 更新了笔记本的内核名称,以更好地反映其用途
Yijun Fu 1 hónapja
szülő
commit
c3e89c588c
1 módosított fájl, 5 hozzáadás és 2 törlés
  1. 5 2
      cuhkpedes/CUHK-PEDES2webdataset.ipynb

+ 5 - 2
cuhkpedes/CUHK-PEDES2webdataset.ipynb

@@ -285,19 +285,22 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 20,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
     "train_images = processed_train_data.shape[0]\n",
+    "print(train_images)\n",
     "train_indexes = list(range(train_images))\n",
     "random.shuffle(train_indexes)\n",
     "\n",
     "test_images = processed_test_data.shape[0]\n",
+    "print(test_images)\n",
     "test_indexes = list(range(test_images))\n",
     "random.shuffle(test_indexes)\n",
     "\n",
     "val_images = processed_val_data.shape[0]\n",
+    "print(val_images)\n",
     "val_indexes = list(range(val_images))\n",
     "random.shuffle(val_indexes)\n",
     "\n",
@@ -383,7 +386,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "Python 3",
+   "display_name": "groupvit",
    "language": "python",
    "name": "python3"
   },