darshanjani commited on
Commit
64eea2e
1 Parent(s): 46f0187

Update Utilities/config.py

Browse files
Files changed (1) hide show
  1. Utilities/config.py +3 -0
Utilities/config.py CHANGED
@@ -31,6 +31,9 @@ TRAIN_MOSAIC_PERCENTAGE = 0.5
31
  TEST_MOSAIC_PERCENTAGE = 0.00
32
  MODEL_STATE_DICT_PATH = os.path.join(MAIN_DIR, "Store/checkpoints/yolov3.pth")
33
 
 
 
 
34
  ANCHORS = [
35
  [(0.28, 0.22), (0.38, 0.48), (0.9, 0.78)],
36
  [(0.07, 0.15), (0.15, 0.11), (0.14, 0.29)],
 
31
  TEST_MOSAIC_PERCENTAGE = 0.00
32
  MODEL_STATE_DICT_PATH = os.path.join(MAIN_DIR, "Store/checkpoints/yolov3.pth")
33
 
34
+ MODEL_CHECKPOINT_PATH = "./Store/epoch=39-step=16560.ckpt"
35
+ EXAMPLE_IMG_PATH = "./Store/examples/"
36
+
37
  ANCHORS = [
38
  [(0.28, 0.22), (0.38, 0.48), (0.9, 0.78)],
39
  [(0.07, 0.15), (0.15, 0.11), (0.14, 0.29)],