Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).
MIT License
1.73k
stars
787
forks
source link
About your Loss version function, and switch it for other #464
Hello, `I studied your code for a long time, but I still don't understand why your loss function has things that I haven't seen in the yolo / yolov2 articles.
In particular, I've noticed that you apply tf.nn.sparse_categorical_ce_from_logits to class_predicted, but I've seen other implementations that simply apply tf.math.softmax to it, then compute the error tf.math.sqrt (predicted_logits - true_logits).
What do you think about this ?
Another question, is if you recommend me to use a loss function like in the following code:
`import numpy as np
import tensorflow as tf
Hello, `I studied your code for a long time, but I still don't understand why your loss function has things that I haven't seen in the yolo / yolov2 articles. In particular, I've noticed that you apply tf.nn.sparse_categorical_ce_from_logits to class_predicted, but I've seen other implementations that simply apply tf.math.softmax to it, then compute the error tf.math.sqrt (predicted_logits - true_logits). What do you think about this ?
Another question, is if you recommend me to use a loss function like in the following code: `import numpy as np import tensorflow as tf
class Loss: def init(self, anchors, lambda_coord=5, lambda_noobj=1): self.anchors = anchors self.lambda_coord = lambda_coord self.lambda_noobj = lambda_noobj self.priors = self.make_priors() self.first_run = True pass
I hope you can help me with this doubt that torments me a lot, thnks!