Spaces:
Sleeping
Sleeping
add compare resnet model
Browse files- 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, (
|
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)
|