Can you tell me if argmax is implemented in SNPE?
self.classes = tf.argmax(self.softmax, axis=3,name='image_output')
what is the equivalent of above line?
Can you tell me if argmax is implemented in SNPE?
Posted: Tue, 2018-02-27 02:27
Forums - Can you tell me if argmax is implemented in SNPE?
Re: Can you tell me if argmax is implemented in SNPE? #1
Posted: Thu, 2018-03-29 21:54
Top
I wasn't able to use ArgMax. As ArgMax usually comes at the end of the neural network. I implemented it outside SNPE after reading data from output tensor. Not a good way but works
Login or Register
to post a comment.
Opinions expressed in the content posted here are the personal opinions of the original authors, and do not necessarily reflect those of Qualcomm Incorporated or its subsidiaries (“Qualcomm”). The content is provided for informational purposes only and is not meant to be an endorsement or representation by Qualcomm or any other party. This site may also provide links or references to non-Qualcomm sites and resources. Qualcomm makes no representations, warranties, or other commitments whatsoever about any non-Qualcomm sites or third-party resources that may be referenced, accessible from, or linked to this site.