Spaces:
Paused
Paused
ajout
Browse files
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.
|
| 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=
|
| 46 |
|
| 47 |
model2 = tf.keras.models.load_model("model/best_ResNet50V2_04_improved_target_augment.keras",compile=False)
|
| 48 |
-
model2_for_gradcam=
|
| 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
|