kaczmarj commited on
Commit
24ac609
1 Parent(s): 135ce4f

Update files to conform to wsinfer-zoo spec

Browse files
Files changed (4) hide show
  1. config.json +33 -18
  2. model.safetensors +3 -0
  3. pytorch_model.bin +3 -0
  4. torchscript_model.pt +2 -2
config.json CHANGED
@@ -1,23 +1,38 @@
1
  {
2
- "num_classes": 3,
3
- "transform": {
4
- "resize_size": 224,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  "mean": [
6
- 0.6462,
7
- 0.507,
8
- 0.8055
9
  ],
10
  "std": [
11
- 0.1381,
12
- 0.1674,
13
- 0.1358
14
  ]
15
- },
16
- "patch_size_pixels": 175,
17
- "spacing_um_px": 0.5,
18
- "class_names": [
19
- "grade3",
20
- "grade4or5",
21
- "benign"
22
- ]
23
- }
 
1
  {
2
+ "spec_version": "1.0",
3
+ "architecture": "resnet34",
4
+ "num_classes": 3,
5
+ "class_names": [
6
+ "grade3",
7
+ "grade4+5",
8
+ "benign"
9
+ ],
10
+ "patch_size_pixels": 175,
11
+ "spacing_um_px": 0.5,
12
+ "transform": [
13
+ {
14
+ "name": "Resize",
15
+ "arguments": {
16
+ "size": 224
17
+ }
18
+ },
19
+ {
20
+ "name": "ToTensor"
21
+ },
22
+ {
23
+ "name": "Normalize",
24
+ "arguments": {
25
  "mean": [
26
+ 0.6462,
27
+ 0.507,
28
+ 0.8055
29
  ],
30
  "std": [
31
+ 0.1381,
32
+ 0.1674,
33
+ 0.1358
34
  ]
35
+ }
36
+ }
37
+ ]
38
+ }
 
 
 
 
 
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f15783d79cb8d1f8dc8004068dc89da6265e189dfc66f82cc0da30cb42fe1842
3
+ size 85232548
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c6948b864e5c232b4a27979252a66484405233d65f72eacf560fef3d41c27785
3
+ size 85284661
torchscript_model.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ea5475460f06461b8d0059d1a848f4e975d0fa25b2b7dd64c514d05365451e35
3
- size 85345884
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:175d1d140886b59ae06af27fef0e7ba760f3920c701fbb5570d7a914a4ecf9c1
3
+ size 85340584