Spaces:
Sleeping
Sleeping
update `coco_url` with AWS native format
Browse files- card_model.py +5 -0
card_model.py
CHANGED
|
@@ -36,6 +36,9 @@ def style():
|
|
| 36 |
"""
|
| 37 |
|
| 38 |
|
|
|
|
|
|
|
|
|
|
| 39 |
def card(img_url, img_w, img_h, boxes):
|
| 40 |
""" This is a hack to streamlit
|
| 41 |
Solution thanks to: https://discuss.streamlit.io/t/display-svg/172/5
|
|
@@ -43,6 +46,7 @@ def card(img_url, img_w, img_h, boxes):
|
|
| 43 |
Also we used the
|
| 44 |
"""
|
| 45 |
_boxes = ""
|
|
|
|
| 46 |
for _id, cx, cy, w, h, label, logit, is_selected, _ in boxes:
|
| 47 |
x, y, w, h = cxywh2xywh(cx, cy, w, h)
|
| 48 |
x = round(img_w * x)
|
|
@@ -80,6 +84,7 @@ def obj_card(img_url, img_w, img_h, cx, cy, w, h, *args, dst_len=100):
|
|
| 80 |
Returns:
|
| 81 |
_obj_html: html string to display object
|
| 82 |
"""
|
|
|
|
| 83 |
w = img_w * w
|
| 84 |
h = img_h * h
|
| 85 |
s = max(w, h)
|
|
|
|
| 36 |
"""
|
| 37 |
|
| 38 |
|
| 39 |
+
def convert_img_url(img_url):
|
| 40 |
+
return img_url.replace('http://images.cocodataset.org', 'https://s3.us-east-1.amazonaws.com/images.cocodataset.org')
|
| 41 |
+
|
| 42 |
def card(img_url, img_w, img_h, boxes):
|
| 43 |
""" This is a hack to streamlit
|
| 44 |
Solution thanks to: https://discuss.streamlit.io/t/display-svg/172/5
|
|
|
|
| 46 |
Also we used the
|
| 47 |
"""
|
| 48 |
_boxes = ""
|
| 49 |
+
img_url = convert_img_url(img_url)
|
| 50 |
for _id, cx, cy, w, h, label, logit, is_selected, _ in boxes:
|
| 51 |
x, y, w, h = cxywh2xywh(cx, cy, w, h)
|
| 52 |
x = round(img_w * x)
|
|
|
|
| 84 |
Returns:
|
| 85 |
_obj_html: html string to display object
|
| 86 |
"""
|
| 87 |
+
img_url = convert_img_url(img_url)
|
| 88 |
w = img_w * w
|
| 89 |
h = img_h * h
|
| 90 |
s = max(w, h)
|