add config.json

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,15 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="2">
+            <item index="0" class="java.lang.String" itemvalue="opencv-python" />
+            <item index="1" class="java.lang.String" itemvalue="opencv-python-headless" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8" project-jdk-type="Python SDK" />
+</project>

.idea/mnist-torch.iml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$" />
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/mnist-torch.iml" filepath="$PROJECT_DIR$/.idea/mnist-torch.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="" vcs="Git" />
+  </component>
+</project>

.idea/workspace.xml

@@ -0,0 +1,39 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ChangeListManager">
+    <list default="true" id="496dc987-855c-4173-8d76-bd94b776a63a" name="Changes" comment="" />
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="MarkdownSettingsMigration">
+    <option name="stateVersion" value="1" />
+  </component>
+  <component name="ProjectId" id="2Os1vawLggIVe2m7oXHeWPnayFD" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent"><![CDATA[{
+  "keyToString": {
+    "RunOnceActivity.OpenProjectViewOnStart": "true",
+    "RunOnceActivity.ShowReadmeOnStart": "true",
+    "last_opened_file_path": "/home/wangjie/modelServing/huggingface/chatglm"
+  }
+}]]></component>
+  <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="application-level" UseSingleDictionary="true" transferred="true" />
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="Default task">
+      <changelist id="496dc987-855c-4173-8d76-bd94b776a63a" name="Changes" comment="" />
+      <created>1682334706298</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1682334706298</updated>
+    </task>
+    <servers />
+  </component>
+</project>

Model.py

@@ -1,20 +0,0 @@
-import torch
-
-class MNIST(torch.nn.Module):
-    def __init__(self):
-        super(MNIST, self).__init__()
-        self.conv = torch.nn.Sequential(torch.nn.Conv2d(1, 32, 3, 1, 1),
-                                        torch.nn.ReLU(),
-                                        torch.nn.Conv2d(32, 64, 3, 1, 1),
-                                        torch.nn.ReLU(),
-                                        torch.nn.MaxPool2d(2, 2))
-        self.dense = torch.nn.Sequential(torch.nn.Linear(14 * 14 * 64, 1024),
-                                         torch.nn.ReLU(),
-                                         torch.nn.Dropout(p=0.2),
-                                         torch.nn.Linear(1024, 10))
-
-    def forward(self, x):
-        x = self.conv(x)
-        x = x.view(-1, 14 * 14 * 64)
-        x = self.dense(x)
-        return x

config.json

@@ -0,0 +1,49 @@
+{
+  "_name_or_path": "google/vit-base-patch16-224-in21k",
+  "architectures": [
+    "ViTForImageClassification"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "encoder_stride": 16,
+  "finetuning_task": "image-classification",
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 768,
+  "id2label": {
+    "0": "0",
+    "1": "1",
+    "2": "2",
+    "3": "3",
+    "4": "4",
+    "5": "5",
+    "6": "6",
+    "7": "7",
+    "8": "8",
+    "9": "9"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "label2id": {
+    "0": "0",
+    "1": "1",
+    "2": "2",
+    "3": "3",
+    "4": "4",
+    "5": "5",
+    "6": "6",
+    "7": "7",
+    "8": "8",
+    "9": "9"
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "vit",
+  "num_attention_heads": 12,
+  "num_channels": 3,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "problem_type": "single_label_classification",
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.22.0.dev0"
+}

inference.py

@@ -1,29 +0,0 @@
-import cv2
-import torch
-from torchvision import datasets, transforms
-import time
-from Model import MNIST
-import numpy
-# from torch.utils.data import DataLoader
-def images2tensor(image):
-    img = cv2.imread(image)
-    img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
-    transf = transforms.ToTensor()
-    img_tensor = torch.unsqueeze(transf(img), dim=0)
-    return img_tensor
-
-
-if __name__ == "__main__":
-    device = torch.device('cpu')
-    model = MNIST().to(device)
-    model.load_state_dict(torch.load('mnist.pkl' , map_location=device))  # load
-    # test_dataset = datasets.MNIST(root='data/', train=False, transform=transforms.ToTensor(), download=True)
-    # test_image , l = test_dataset[0]
-    input_data = images2tensor("0.png")
-    start = time.time()
-    res = model(input_data)
-    end = time.time()
-    res = res.detach().numpy()
-    # res = numpy.array(res)
-    print("手写数字图片检测的结果为：", res.argmax())
-    # print("infer time: ", end - start)

mnist.pkl → pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f23ce66930c4c38a18e1b51ab4306e2c70319bb3ce7ef1820bf3390876e0b751
-size 51503055
+oid sha256:354112314c2a6a219b585b09197af8a67b09d110ba53b09d8cfdf8a93a2bbf1c
+size 343291569