Update config.json

HichTala · Feb 9, 2024 · 8dfb847 · 8dfb847
1 parent e3a2270
commit 8dfb847
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/config.json b/config.json
@@ -1,6 +1,7 @@
 {
   "trained_models": "/models",
   "data_path": "/data",
+  "webcam": 0,
   "card_types": ["spell", "monster", "normal_monster", "trap", "fusion", "ritual"],
   "monster": {
     "colors": 0,

diff --git a/draw.py b/draw.py
@@ -28,6 +28,9 @@ def __init__(self, config, deck_list, source, debug=False):
             device=device
         )
 
+        if source == '0':
+            source = str(self.configs['webcam'])
+
         self.results = model_regression(
             source=source,
             show_labels=False,