Upload folder using huggingface_hub

models/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ee6f0b0f6d957c868e1fb383c627d0de3095f5fac670c084e81cb81b29b43b73
+oid sha256:6d5e8017d418b955cc16dd7a3f4ae86cc15f9be6df35bdbd2ab275583bf05b60
 size 1074051192

notebooks/nobg.ipynb

@@ -2,9 +2,18 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 11,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "c:\\Users\\hiroga\\miniconda3\\envs\\pokemon-pal\\Lib\\site-packages\\torchvision\\transforms\\functional.py:1603: UserWarning: The default value of the antialias parameter of all the resizing transforms (Resize(), RandomResizedCrop(), etc.) will change from None to True in v0.17, in order to be consistent across the PIL and Tensor backends. To suppress this warning, directly pass antialias=True (recommended, future default), antialias=None (current default, which means False for Tensors and True for PIL), or antialias=False (only works on Tensors - PIL will still use antialiasing). This also applies if you are using the inference transforms from the models weights: update the call to weights.transforms(antialias=True).\n",
+      "  warnings.warn(\n"
+     ]
+    }
+   ],
    "source": [
     "import torch\n",
     "from carvekit.api.high import HiInterface\n",
@@ -24,8 +33,8 @@
     "\n",
     "# input_dir = \"../data/raw\"\n",
     "# output_dir = \"../data/nobg\"\n",
-    "input_dir = \"../data/raw/ポケットモンスターシールド\"\n",
-    "output_dir = \"../data/nobg/ポケットモンスターシールド\"\n",
+    "input_dir = \"../data/raw/paldex.io\"\n",
+    "output_dir = \"../data/nobg/paldex.io\"\n",
     "\n",
     "# Create output directory if it doesn't exist\n",
     "os.makedirs(output_dir, exist_ok=True)\n",

notebooks/train.ipynb

@@ -47,7 +47,8 @@
     "]\n",
     "pal_dir: list[str] = [\n",
     "    f\"{data_dir}/#パルワールド/\", \n",
-    "    f\"{data_dir}/every-pal-in-palworld-a-complete-paldeck-list/\"\n",
+    "    f\"{data_dir}/every-pal-in-palworld-a-complete-paldeck-list/\",\n",
+    "    f\"{data_dir}/paldex.io/\"\n",
     "]\n",
     "\n",
     "pokemon_images: list[str] = [os.path.join(dir, file) for dir in pokemon_dir for file in os.listdir(dir)]\n",
@@ -198,10 +199,11 @@
     "        \n",
     "    # Save the model\n",
     "    model_dir = '../models/'\n",
-    "    if not os.path.exists(model_dir):\n",
-    "        os.makedirs(model_dir)\n",
+    "    snapshots_dir = f\"{model_dir}snapshots/\"\n",
+    "    if not os.path.exists(snapshots_dir):\n",
+    "        os.makedirs(snapshots_dir)\n",
     "    tensors = {name: param for name, param in model.named_parameters()}\n",
-    "    save_file(tensors, f\"{model_dir}SimpleCNN_{epoch+1}_{data_dir.replace('.', '').replace('/', '_')}_{image_size}x{image_size}.safetensors\")\n",
+    "    save_file(tensors, f\"{snapshots_dir}SimpleCNN_{epoch+1}_{data_dir.replace('.', '').replace('/', '_')}_{image_size}x{image_size}.safetensors\")\n",
     "    save_file(tensors, f\"{model_dir}model.safetensors\")\n",
     "\n",
     "    # Plotting the losses\n",