Spaces:
Runtime error
Runtime error
Commit ·
e799435
1
Parent(s): 1735afc
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,8 @@ examples = [["sample_images/"+file] for file in files]
|
|
| 50 |
def show_preds(input_image, display_label, display_bbox, detection_threshold):
|
| 51 |
if detection_threshold==0: detection_threshold=0.5
|
| 52 |
img = PIL.Image.fromarray(input_image, 'RGB')
|
| 53 |
-
pred_dict = model_type.end2end_detect(img, valid_tfms,
|
|
|
|
| 54 |
display_label=display_label, display_bbox=display_bbox, return_img=True,
|
| 55 |
font_size=16, label_color="#FF59D6")
|
| 56 |
return pred_dict['img']
|
|
|
|
| 50 |
def show_preds(input_image, display_label, display_bbox, detection_threshold):
|
| 51 |
if detection_threshold==0: detection_threshold=0.5
|
| 52 |
img = PIL.Image.fromarray(input_image, 'RGB')
|
| 53 |
+
pred_dict = model_type.end2end_detect(img, valid_tfms, learn, ##model,
|
| 54 |
+
class_map=class_map, detection_threshold=detection_threshold,
|
| 55 |
display_label=display_label, display_bbox=display_bbox, return_img=True,
|
| 56 |
font_size=16, label_color="#FF59D6")
|
| 57 |
return pred_dict['img']
|