1aurent commited on
Commit
732166c
1 Parent(s): f1e9369

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +11 -11
config.json CHANGED
@@ -3,28 +3,28 @@
3
  "num_classes": 2,
4
  "num_features": 1280,
5
  "pretrained_cfg": {
6
- "tag": "ra_in1k",
7
  "custom_load": false,
8
  "input_size": [
9
  3,
10
- 224,
11
- 224
12
  ],
13
  "fixed_input_size": false,
14
  "interpolation": "bicubic",
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
 
3
  "num_classes": 2,
4
  "num_features": 1280,
5
  "pretrained_cfg": {
6
+ "tag": "tiatoolbox-pcam",
7
  "custom_load": false,
8
  "input_size": [
9
  3,
10
+ 96,
11
+ 96
12
  ],
13
  "fixed_input_size": false,
14
  "interpolation": "bicubic",
15
+ "crop_pct": 1.0,
16
  "crop_mode": "center",
17
  "mean": [
18
+ 0.0,
19
+ 0.0,
20
+ 0.0
21
  ],
22
  "std": [
23
+ 1.0,
24
+ 1.0,
25
+ 1.0
26
  ],
27
+ "num_classes": 2,
28
  "pool_size": [
29
  7,
30
  7