Add model
Browse files- config.json +4 -4
config.json
CHANGED
|
@@ -3,8 +3,8 @@
|
|
| 3 |
"num_classes": 2,
|
| 4 |
"input_size": [
|
| 5 |
3,
|
| 6 |
-
|
| 7 |
-
|
| 8 |
],
|
| 9 |
"pool_size": [
|
| 10 |
7,
|
|
@@ -27,8 +27,8 @@
|
|
| 27 |
"architecture": "resnet50",
|
| 28 |
"num_features": 2048,
|
| 29 |
"labels": [
|
| 30 |
-
"
|
| 31 |
-
"
|
| 32 |
],
|
| 33 |
"label_names": [
|
| 34 |
"Gen",
|
|
|
|
| 3 |
"num_classes": 2,
|
| 4 |
"input_size": [
|
| 5 |
3,
|
| 6 |
+
512,
|
| 7 |
+
512
|
| 8 |
],
|
| 9 |
"pool_size": [
|
| 10 |
7,
|
|
|
|
| 27 |
"architecture": "resnet50",
|
| 28 |
"num_features": 2048,
|
| 29 |
"labels": [
|
| 30 |
+
"Gen",
|
| 31 |
+
"Real"
|
| 32 |
],
|
| 33 |
"label_names": [
|
| 34 |
"Gen",
|