Spaces:
Sleeping
Sleeping
cyberai-1 commited on
Commit ·
5375ee4
1
Parent(s): 17f6e62
Relax unknown prediction rejection
Browse files
app.py
CHANGED
|
@@ -26,10 +26,10 @@ DATABASE_PATH = BASE_DIR / "instance" / "traffic_signs.sqlite3"
|
|
| 26 |
MODEL_PATH = BASE_DIR / "traffic_classifier.h5"
|
| 27 |
ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg", "webp"}
|
| 28 |
NORMALIZE_INPUT = os.environ.get("NORMALIZE_INPUT", "false").lower() in {"1", "true", "yes"}
|
| 29 |
-
MC_DROPOUT_PASSES = int(os.environ.get("MC_DROPOUT_PASSES", "
|
| 30 |
-
UNKNOWN_CONFIDENCE_THRESHOLD = float(os.environ.get("UNKNOWN_CONFIDENCE_THRESHOLD", "0.
|
| 31 |
-
UNKNOWN_MARGIN_THRESHOLD = float(os.environ.get("UNKNOWN_MARGIN_THRESHOLD", "0.
|
| 32 |
-
UNKNOWN_VOTE_THRESHOLD = float(os.environ.get("UNKNOWN_VOTE_THRESHOLD", "0.
|
| 33 |
URL_CONTENT_TYPES = {
|
| 34 |
"image/jpeg": "jpg",
|
| 35 |
"image/png": "png",
|
|
|
|
| 26 |
MODEL_PATH = BASE_DIR / "traffic_classifier.h5"
|
| 27 |
ALLOWED_EXTENSIONS = {"png", "jpg", "jpeg", "webp"}
|
| 28 |
NORMALIZE_INPUT = os.environ.get("NORMALIZE_INPUT", "false").lower() in {"1", "true", "yes"}
|
| 29 |
+
MC_DROPOUT_PASSES = int(os.environ.get("MC_DROPOUT_PASSES", "1"))
|
| 30 |
+
UNKNOWN_CONFIDENCE_THRESHOLD = float(os.environ.get("UNKNOWN_CONFIDENCE_THRESHOLD", "0.999"))
|
| 31 |
+
UNKNOWN_MARGIN_THRESHOLD = float(os.environ.get("UNKNOWN_MARGIN_THRESHOLD", "0.0"))
|
| 32 |
+
UNKNOWN_VOTE_THRESHOLD = float(os.environ.get("UNKNOWN_VOTE_THRESHOLD", "0.0"))
|
| 33 |
URL_CONTENT_TYPES = {
|
| 34 |
"image/jpeg": "jpg",
|
| 35 |
"image/png": "png",
|