Spaces:

stevfoy
/

Turfgrass_Divots

Runtime error

stevfoy commited on Nov 4, 2023

Commit

9222fe1

1 Parent(s): 01869fd

extra

Files changed (2) hide show

app.py CHANGED Viewed

@@ -30,7 +30,7 @@ import gradio as gr
 import os
 model = YOLOv3Lightning()
-model.load_state_dict(torch.load("yolov3_model.pth", map_location=torch.device('cpu')), strict=False)
 model.setup(stage="test")
 IMAGE_SIZE = 416

 import os
 model = YOLOv3Lightning()
+model.load_state_dict(torch.load("yolov3_608_ckpt_40.pth", map_location=torch.device('cpu')), strict=False)
 model.setup(stage="test")
 IMAGE_SIZE = 416

config.py CHANGED Viewed

@@ -10,7 +10,7 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 # seed_everything()  # If you want deterministic behavior
 NUM_WORKERS = 4
 BATCH_SIZE = 16
-IMAGE_SIZE = 608
 NUM_CLASSES = 2
 LEARNING_RATE = 1e-5
 WEIGHT_DECAY = 1e-4

 # seed_everything()  # If you want deterministic behavior
 NUM_WORKERS = 4
 BATCH_SIZE = 16
+IMAGE_SIZE = 416
 NUM_CLASSES = 2
 LEARNING_RATE = 1e-5
 WEIGHT_DECAY = 1e-4