Commit
•
35aa782
1
Parent(s):
ac8f8dc
Fix label
Browse files- README.md +2 -2
- output.jpg +0 -0
- run_sam_rknn.py +1 -1
README.md
CHANGED
|
@@ -21,7 +21,7 @@ license: unknown
|
|
| 21 |
|
| 22 |
提示：
|
| 23 |
|
| 24 |
-
`{"type": "point", "data": [540, 512], "label":
|
| 25 |
|
| 26 |
输出:
|
| 27 |
|
|
@@ -61,7 +61,7 @@ Input:
|
|
| 61 |
|
| 62 |
Prompt:
|
| 63 |
|
| 64 |
-
`{"type": "point", "data": [540, 512], "label":
|
| 65 |
|
| 66 |
Output:
|
| 67 |
|
|
|
|
| 21 |
|
| 22 |
提示：
|
| 23 |
|
| 24 |
+
`{"type": "point", "data": [540, 512], "label": 1}`
|
| 25 |
|
| 26 |
输出:
|
| 27 |
|
|
|
|
| 61 |
|
| 62 |
Prompt:
|
| 63 |
|
| 64 |
+
`{"type": "point", "data": [540, 512], "label": 1}`
|
| 65 |
|
| 66 |
Output:
|
| 67 |
|
output.jpg
CHANGED
|
|
run_sam_rknn.py
CHANGED
|
@@ -204,7 +204,7 @@ if __name__ == "__main__":
|
|
| 204 |
image = cv2.imread("input.jpg")
|
| 205 |
embedding = segmenter.encode(image)
|
| 206 |
prompt = [
|
| 207 |
-
{"type": "point", "data": [540, 512], "label":
|
| 208 |
]
|
| 209 |
masks = segmenter.predict_masks(embedding, prompt)
|
| 210 |
print(masks.shape)
|
|
|
|
| 204 |
image = cv2.imread("input.jpg")
|
| 205 |
embedding = segmenter.encode(image)
|
| 206 |
prompt = [
|
| 207 |
+
{"type": "point", "data": [540, 512], "label": 1},
|
| 208 |
]
|
| 209 |
masks = segmenter.predict_masks(embedding, prompt)
|
| 210 |
print(masks.shape)
|