1aurent commited on
Commit
7db91b9
1 Parent(s): cdd392f

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +20 -14
config.json CHANGED
@@ -3,35 +3,41 @@
3
  "num_classes": 2,
4
  "num_features": 2048,
5
  "pretrained_cfg": {
6
- "tag": "tv_in1k",
7
  "custom_load": false,
8
  "input_size": [
9
  3,
10
- 224,
11
- 224
 
 
 
 
 
12
  ],
13
  "fixed_input_size": false,
14
- "interpolation": "bilinear",
15
- "crop_pct": 0.875,
 
16
  "crop_mode": "center",
17
  "mean": [
18
- 0.485,
19
- 0.456,
20
- 0.406
21
  ],
22
  "std": [
23
- 0.229,
24
- 0.224,
25
- 0.225
26
  ],
27
- "num_classes": 1000,
28
  "pool_size": [
29
  7,
30
  7
31
  ],
32
  "first_conv": "conv1",
33
  "classifier": "fc",
34
- "license": "bsd-3-clause",
35
- "origin_url": "https://github.com/pytorch/vision"
36
  }
37
  }
 
3
  "num_classes": 2,
4
  "num_features": 2048,
5
  "pretrained_cfg": {
6
+ "tag": "tiatoolbox-pcam",
7
  "custom_load": false,
8
  "input_size": [
9
  3,
10
+ 96,
11
+ 96
12
+ ],
13
+ "test_input_size": [
14
+ 3,
15
+ 96,
16
+ 96
17
  ],
18
  "fixed_input_size": false,
19
+ "interpolation": "bicubic",
20
+ "crop_pct": 1.0,
21
+ "test_crop_pct": 1.0,
22
  "crop_mode": "center",
23
  "mean": [
24
+ 0.0,
25
+ 0.0,
26
+ 0.0
27
  ],
28
  "std": [
29
+ 1.0,
30
+ 1.0,
31
+ 1.0
32
  ],
33
+ "num_classes": 2,
34
  "pool_size": [
35
  7,
36
  7
37
  ],
38
  "first_conv": "conv1",
39
  "classifier": "fc",
40
+ "origin_url": "https://github.com/huggingface/pytorch-image-models",
41
+ "paper_ids": "arXiv:2110.00476"
42
  }
43
  }