cyberai-1 commited on
Commit
5375ee4
·
1 Parent(s): 17f6e62

Relax unknown prediction rejection

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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", "3"))
30
- UNKNOWN_CONFIDENCE_THRESHOLD = float(os.environ.get("UNKNOWN_CONFIDENCE_THRESHOLD", "0.50"))
31
- UNKNOWN_MARGIN_THRESHOLD = float(os.environ.get("UNKNOWN_MARGIN_THRESHOLD", "0.10"))
32
- UNKNOWN_VOTE_THRESHOLD = float(os.environ.get("UNKNOWN_VOTE_THRESHOLD", "0.60"))
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",