Spaces:
Sleeping
Sleeping
updte xai sampling size
Browse files- inference_resnet_v2.py +1 -1
inference_resnet_v2.py
CHANGED
@@ -53,7 +53,7 @@ def inference_resnet_embedding_v2(x,model,size=384,n_classes=140,n_top=10):
|
|
53 |
|
54 |
|
55 |
x = tf.image.resize(x, (size, size))
|
56 |
-
x = tf.reshape(x, (
|
57 |
embedding = model.predict(np.array([x]))[0][0]
|
58 |
|
59 |
|
|
|
53 |
|
54 |
|
55 |
x = tf.image.resize(x, (size, size))
|
56 |
+
x = tf.reshape(x, (384, 384, 3))/255
|
57 |
embedding = model.predict(np.array([x]))[0][0]
|
58 |
|
59 |
|