Spaces:
Sleeping
Sleeping
Commit
·
a4f243f
1
Parent(s):
848cadc
Minor change
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ data = "data.yaml"
|
|
20 |
# Load model
|
21 |
device = select_device('cpu')
|
22 |
model = DetectMultiBackend(weights, device=device, dnn=False, data=data, fp16=False)
|
23 |
-
target_layers = model.model.model[-1]
|
24 |
|
25 |
false_detection_data = glob(os.path.join("false_detection", '*.jpg'))
|
26 |
false_detection_data = [x.replace('\\', '/') for x in false_detection_data]
|
|
|
20 |
# Load model
|
21 |
device = select_device('cpu')
|
22 |
model = DetectMultiBackend(weights, device=device, dnn=False, data=data, fp16=False)
|
23 |
+
target_layers = [model.model.model[-1]]
|
24 |
|
25 |
false_detection_data = glob(os.path.join("false_detection", '*.jpg'))
|
26 |
false_detection_data = [x.replace('\\', '/') for x in false_detection_data]
|