andy-wyx commited on
Commit
5852c4b
·
1 Parent(s): 1c118fd

add compare resnet model

Browse files
Files changed (1) hide show
  1. inference_resnet_v2.py +1 -1
inference_resnet_v2.py CHANGED
@@ -62,7 +62,7 @@ def inference_resnet_embedding_v2(x,model,size=384,n_classes=140,n_top=10):
62
  def inference_resnet_finer_v2(x,model,size=384,n_classes=142,n_top=10):
63
 
64
  x = tf.image.resize(x, (size, size))
65
- x = tf.reshape(x, (-1, 384, 384, 3))/255
66
  # _, batch_logits = model.predict(x)
67
  # predictions = tf.math.top_k(batch_logits, k=10)
68
  # print(predictions)
 
62
  def inference_resnet_finer_v2(x,model,size=384,n_classes=142,n_top=10):
63
 
64
  x = tf.image.resize(x, (size, size))
65
+ x = tf.reshape(x, (384, 384, 3))/255
66
  # _, batch_logits = model.predict(x)
67
  # predictions = tf.math.top_k(batch_logits, k=10)
68
  # print(predictions)