|
|
|
@ -254,7 +254,7 @@ def train_onet(model_store_path, end_epoch,imdb,
|
|
|
|
|
show4 = landmark_loss.data.tolist()[0]
|
|
|
|
|
show5 = all_loss.data.tolist()[0]
|
|
|
|
|
|
|
|
|
|
print "%s : Epoch: %d, Step: %d, accuracy: %s, det loss: %s, bbox loss: %s, landmark loss: %s, all_loss: %s, lr:%s "%(datetime.datetime.now(),cur_epoch,batch_idx, show1,show2,show3,show4,show5,base_lr)
|
|
|
|
|
print("%s : Epoch: %d, Step: %d, accuracy: %s, det loss: %s, bbox loss: %s, landmark loss: %s, all_loss: %s, lr:%s "%(datetime.datetime.now(),cur_epoch,batch_idx, show1,show2,show3,show4,show5,base_lr))
|
|
|
|
|
accuracy_list.append(accuracy)
|
|
|
|
|
cls_loss_list.append(cls_loss)
|
|
|
|
|
bbox_loss_list.append(box_offset_loss)
|
|
|
|
|