rkonan commited on
Commit
58919ac
·
1 Parent(s): 8c4cb40
app/model.py CHANGED
@@ -28,7 +28,7 @@ logging.basicConfig(
28
  format="%(asctime)s [%(levelname)s] %(name)s: %(message)s"
29
  )
30
  logger = logging.getLogger(__name__)
31
- confidence_threshold=0.6
32
  entropy_threshold=1.5
33
 
34
  class ModelStruct(TypedDict):
@@ -42,10 +42,10 @@ class ModelStruct(TypedDict):
42
 
43
  def load_models() -> list[ModelStruct]:
44
  model1 = tf.keras.models.load_model("model/best_EfficientNetV2M_03_improved_global_augment.keras",compile=False)
45
- model1_for_gradcam=tf.keras.models.load_model("model/best_efficientnetv2m_gradcam.keras",compile=False)
46
 
47
  model2 = tf.keras.models.load_model("model/best_ResNet50V2_04_improved_target_augment.keras",compile=False)
48
- model2_for_gradcam=tf.keras.models.load_model("model/best_ResNet50V2_gradcam.keras",compile=False)
49
 
50
  return [
51
 
 
28
  format="%(asctime)s [%(levelname)s] %(name)s: %(message)s"
29
  )
30
  logger = logging.getLogger(__name__)
31
+ confidence_threshold=0.5
32
  entropy_threshold=1.5
33
 
34
  class ModelStruct(TypedDict):
 
42
 
43
  def load_models() -> list[ModelStruct]:
44
  model1 = tf.keras.models.load_model("model/best_EfficientNetV2M_03_improved_global_augment.keras",compile=False)
45
+ model1_for_gradcam=model1
46
 
47
  model2 = tf.keras.models.load_model("model/best_ResNet50V2_04_improved_target_augment.keras",compile=False)
48
+ model2_for_gradcam=model2
49
 
50
  return [
51
 
model/best_ResNet50V2_gradcam.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3a850596cb72e8801b067d256de18823f5740629f3e7f39e0911fdc4f8cdc2a1
3
+ size 97195984
model/best_efficientnetv2m_gradcam.keras ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:303b262f1a720ec476e5521f0728843c90650dabbe0adfc7c7a5a33e6fd9e9d2
3
+ size 216600984