Commit Data paths / Weight Paths / Parameters / Hyper-Paramaters

backend/ImageSearch.ipynb

+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "# **Data paths / Weight Paths / Parameters / Hyper-Paramaters**"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {
+    "vscode": {
+     "languageId": "plaintext"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "data_dir = 'data/'\n",
+    "\n",
+    "seed = 1234\n",
+    "\n",
+    "category_dict = {\n",
+    "            'computer' : 0,\n",
+    "            'fashion' : 1,\n",
+    "            'furniture' : 2\n",
+    "                }\n",
+    "\n",
+    "sub_category_dict = {\n",
+    "                'Keyboard' : 0,              #computer\n",
+    "                'Mouse' : 1,                 #computer\n",
+    "                'Printer' : 2,               #computer\n",
+    "                'Web Camera' : 3,            #computer\n",
+    "                'Casual Shoes' : 4,            #fashion\n",
+    "                'Formal Shoes' : 5,            #fashion\n",
+    "                'Frocks' : 6,                  #fashion\n",
+    "                'Jeans' : 7,                   #fashion\n",
+    "                'Sandals' : 8,                 #fashion\n",
+    "                'Shirts' : 9,                  #fashion\n",
+    "                'Slipper' : 10,                 #fashion\n",
+    "                'Sport Shoes' : 11,             #fashion\n",
+    "                'T-Shirts' : 12,                #fashion\n",
+    "                'bed' : 13,                   #furniture\n",
+    "                'chair' : 14,                 #furniture\n",
+    "                'sofa' : 15,                  #furniture\n",
+    "                'swivelchair' : 16,           #furniture\n",
+    "                'table' : 17                  #furniture\n",
+    "                }\n",
+    "\n",
+    "\n",
+    "input_shape = (3, 224, 224)\n",
+    "input_size = (224, 224)\n",
+    "train_size = 0.85\n",
+    "\n",
+    "learning_rate = 0.0001\n",
+    "batch_size_train = 8\n",
+    "batch_size_test = 6\n",
+    "epoches = 20\n",
+    "\n",
+    "WEIGHT_PATH = 'weights/ImageSearch/BaseCNN.pt'\n",
+    "FEATURE_PATH = 'weights/ImageSearch/Features.pt'\n",
+    "FEATURE_MODEL_WEIGHT_PATH = 'weights/ImageSearch/FeatureExtractorCNN.pt'"
+   ]
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  },
+  "orig_nbformat": 4
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}