Hugging Face's logo

Spaces:
Akechi93
/
fire-detection-system
Runtime error

App Files Community
Akechi93 commited on
Commit
9b2a3be
·
verified ·
1 Parent(s): 3fc81ec

Update model_info.json

Browse files
Files changed (1) hide show
  1. model_info.json +8 -2
model_info.json CHANGED
@@ -1,10 +1,16 @@
1
  {
2
  "model_type": "ResNet-34",
3
  "num_classes": 2,
4
- "input_size": [224, 224],
 
 
 
5
  "final_test_accuracy": 0.8801,
6
  "training_epochs": 20,
7
- "class_names": ["Non-Fire", "Fire"],
 
 
 
8
  "dataset_info": {
9
  "train_dataset": "fire_dataset (18 images)",
10
  "test_dataset": "Our_Dataset_610 (610 images)",
 
1
  {
2
  "model_type": "ResNet-34",
3
  "num_classes": 2,
4
+ "input_size": [
5
+ 224,
6
+ 224
7
+ ],
8
  "final_test_accuracy": 0.8801,
9
  "training_epochs": 20,
10
+ "class_names": [
11
+ "Fire",
12
+ "Non-Fire"
13
+ ],
14
  "dataset_info": {
15
  "train_dataset": "fire_dataset (18 images)",
16
  "test_dataset": "Our_Dataset_610 (610 images)",