We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c1e3dfa commit 8ce6a13Copy full SHA for 8ce6a13
tests/test_simba.py
@@ -13,15 +13,15 @@ class TestSimBA(unittest.TestCase):
13
14
def test_simba(self):
15
# Load Image [0.0, 1.0]
16
- x = np.asarray(Image.open("tests/dog.jpg").resize((32, 32))) / 255.0
+ x = np.asarray(Image.open("tests/cat.jpg").resize((32, 32))) / 255.0
17
18
# Initialize API Model
19
model = VGG16Cifar10("https://api.wuhanstudio.uk" + "/vgg16_cifar10")
20
21
# Get Preditction
22
y_pred = model.predict(np.array([x]))[0]
23
24
- assert (np.argmax(y_pred) == 5)
+ assert (np.argmax(y_pred) == 3)
25
26
# SimBA Attack
27
simba = SimBA(model)
0 commit comments