摘要: 1. 加载model及训练权重 self.num_classes = len(self.class_names) + 1 self.model = SSD300((300,300,3), self.num_classes) self.model.load_weights('ssd_model.h5' 阅读全文
posted @ 2020-12-15 23:47 learningcaiji 阅读(1563) 评论(0) 推荐(0) 编辑
摘要: SSD损失分为两部分,类别损失和回归框位置损失 其中,类别损失采用softmax损失,回顾框损失采用l1——smooth损失。 1. softmax损失: def _softmax_loss(self, y_true, y_pred): y_pred = tf.maximum(y_pred, 1e- 阅读全文
posted @ 2020-12-15 22:55 learningcaiji 阅读(749) 评论(0) 推荐(0) 编辑