andy-wyx commited on
Commit
f71ea6c
·
1 Parent(s): 8bf9677

updte xai sampling size

Browse files
Files changed (1) hide show
  1. 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, (-1, 384, 384, 3))/255
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