ssd_tensorflow_traffic_sign_detection
ssd_tensorflow_traffic_sign_detection copied to clipboard
error
I couldn't able to install Pickle on python 3 and also I am getting the following error. Please help me to resolve this.
Find the aolution here.
well If you are using tensorflow >1.1.0 you will have to do the following changes:
-
Line 60 of model.py from
conf_loss = tf.nn.sparse_softmax_cross_entropy_with_logits(logits, y_true_conf)
toconf_loss = tf.nn.sparse_softmax_cross_entropy_with_logits(logits=logits, labels=y_true_conf)
-
Line 71 of model.py from
loc_loss = tf.select(smooth_l1_condition, loc_loss_l2, loc_loss_l1)
toloc_loss = tf.where(smooth_l1_condition, loc_loss_l2, loc_loss_l1)
-
Line 164 and 165 of model.py from
final_pred_conf = tf.concat(1, preds_conf)
final_pred_loc = tf.concat(1, preds_loc)
tofinal_pred_conf = tf.concat(preds_conf, 1)
final_pred_loc = tf.concat(preds_loc, 1)
and the inference.py should run
@YashBansod thank you, that helped!