Add files to conform to wsinfer-zoo schema
Browse files- config.json +1 -36
- model.safetensors +3 -0
- pytorch_model.bin +3 -0
- torchscript_model.pt +2 -2
config.json
CHANGED
@@ -1,36 +1 @@
|
|
1 |
-
{
|
2 |
-
"architecture": "resnet34",
|
3 |
-
"num_classes": 2,
|
4 |
-
"class_names": [
|
5 |
-
"notumor",
|
6 |
-
"tumor"
|
7 |
-
],
|
8 |
-
"patch_size_pixels": 350,
|
9 |
-
"spacing_um_px": 0.25,
|
10 |
-
"transform": [
|
11 |
-
{
|
12 |
-
"name": "Resize",
|
13 |
-
"arguments": {
|
14 |
-
"size": 224
|
15 |
-
}
|
16 |
-
},
|
17 |
-
{
|
18 |
-
"name": "ToTensor"
|
19 |
-
},
|
20 |
-
{
|
21 |
-
"name": "Normalize",
|
22 |
-
"arguments": {
|
23 |
-
"mean": [
|
24 |
-
0.7238,
|
25 |
-
0.5716,
|
26 |
-
0.6779
|
27 |
-
],
|
28 |
-
"std": [
|
29 |
-
0.112,
|
30 |
-
0.1459,
|
31 |
-
0.1089
|
32 |
-
]
|
33 |
-
}
|
34 |
-
}
|
35 |
-
]
|
36 |
-
}
|
|
|
1 |
+
{"spec_version": "1.0", "architecture": "resnet34", "num_classes": 2, "class_names": ["notumor", "tumor"], "patch_size_pixels": 350, "spacing_um_px": 0.25, "transform": [{"name": "Resize", "arguments": {"size": 224}}, {"name": "ToTensor"}, {"name": "Normalize", "arguments": {"mean": [0.7238, 0.5716, 0.6779], "std": [0.112, 0.1459, 0.1089]}}]}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
model.safetensors
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:49e47bb52e60694b351a5315a4156875d42b2602adcb226f5a0fe03a5aee878c
|
3 |
+
size 85230496
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:7cb91ab65093080ad2adffee139db476b1c73b7371144e1bcd79be6cba80fe38
|
3 |
+
size 85282613
|
torchscript_model.pt
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
-
size
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:663d25424a7ef7390a260c48c741e516893983dd203824f5844279fa8811c941
|
3 |
+
size 85337454
|