search

chenjun2hao / CenterFace.pytorch

unofficial version of centerface, which achieves the best balance between speed and accuracy at face detection
236 stars 60 forks source link

loss nan #27

Open alicera opened 4 years ago

alicera commented 4 years ago

python main.py I dont change the setting to train

4 |ETA: 0:05:49 |loss nan |hm_loss nan |lm_loss nan |wh_loss nan |off_loss nanmulti_pose/dla |########### | train: [8][546/1583]|Tot: 0:03:05 |ETA: 0:05:49 |loss nan |hm_loss nan |lm_loss nan |wh_loss nan |off_lo

how to resize your images size for training

iytChen commented 4 years ago

same question， anyone knows how to sovle this problem?

YangYangGirl commented 4 years ago

https://github.com/chenjun2hao/CenterFace.pytorch/blob/37a351be07606136561bc56eae716f8575b61e00/src/lib/trains/multi_pose.py#L34

change to this: if not opt.mse_loss: output['hm'] = _sigmoid(output['hm'])