Spaces:
Runtime error
Runtime error
NMPhap
commited on
Commit
•
4ad0e47
1
Parent(s):
a485c33
<Fix>: fix test
Browse files
app/custom_mmcv/test_custom_mmcv.py
CHANGED
|
@@ -30,22 +30,22 @@ class TestCustomMMCV():
|
|
| 30 |
image = mmcv.imread('demo.jpg')
|
| 31 |
bboxes = np.ones((1,5))
|
| 32 |
labels = np.zeros(1, dtype=np.int32)
|
| 33 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red", text_color='red'
|
| 34 |
assert (result[1,1,:] == (0,0,255)).all()
|
| 35 |
with pytest.raises(AssertionError):
|
| 36 |
bboxes = np.ones((1,3))
|
| 37 |
labels = np.zeros(1, dtype=np.int32)
|
| 38 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
| 39 |
with pytest.raises(AssertionError):
|
| 40 |
bboxes = np.ones((1,7))
|
| 41 |
labels = np.zeros(1, dtype=np.int32)
|
| 42 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
| 43 |
with pytest.raises(AssertionError):
|
| 44 |
bboxes = np.ones((1,5))
|
| 45 |
labels = np.zeros(4, dtype=np.int32)
|
| 46 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
| 47 |
with pytest.raises(AssertionError):
|
| 48 |
bboxes = np.ones((2,5))
|
| 49 |
labels = np.zeros(1, dtype=np.int32)
|
| 50 |
-
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red"
|
| 51 |
|
|
|
|
| 30 |
image = mmcv.imread('demo.jpg')
|
| 31 |
bboxes = np.ones((1,5))
|
| 32 |
labels = np.zeros(1, dtype=np.int32)
|
| 33 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red", text_color='red')
|
| 34 |
assert (result[1,1,:] == (0,0,255)).all()
|
| 35 |
with pytest.raises(AssertionError):
|
| 36 |
bboxes = np.ones((1,3))
|
| 37 |
labels = np.zeros(1, dtype=np.int32)
|
| 38 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
| 39 |
with pytest.raises(AssertionError):
|
| 40 |
bboxes = np.ones((1,7))
|
| 41 |
labels = np.zeros(1, dtype=np.int32)
|
| 42 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
| 43 |
with pytest.raises(AssertionError):
|
| 44 |
bboxes = np.ones((1,5))
|
| 45 |
labels = np.zeros(4, dtype=np.int32)
|
| 46 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
| 47 |
with pytest.raises(AssertionError):
|
| 48 |
bboxes = np.ones((2,5))
|
| 49 |
labels = np.zeros(1, dtype=np.int32)
|
| 50 |
+
result = imshow_det_bboxes(image, bboxes, labels, class_names=classNames, bbox_color="red",text_color="red")
|
| 51 |
|