Tensorflow-Fast-MPNCOV icon indicating copy to clipboard operation
Tensorflow-Fast-MPNCOV copied to clipboard

Published 20 hours ago • verified publisher icon XuChunqiao

Predict images

Open harshsp31 opened this issue 4 years ago • 1 comments

Hi @XuChunqiao ,

Thanks for sharing your amazing work. Could you help me by letting me know how to predict an image using the fine tuned model?

Regards

harshsp31 avatar Jun 26 '20 16:06 harshsp31

Sorry for the late reply. If you use single GPU, you can directly call function test_step(self, inputs) in main_multi_GPU.py LIne128, and the variable predictions is the score vector of each class.Then utilize function tf.argmax(predictions, dim=0) you can get the index of maximum

XuChunqiao avatar Oct 16 '20 07:10 XuChunqiao