search

zlzeng / DeepFloorplan

GNU General Public License v3.0
270 stars 103 forks source link

The blanced entropy loss can't work #12

Open exogeny opened 4 years ago

exogeny commented 4 years ago

I run the training code in r3d dataset, and can't get any room used the trained model. The EVAL_pretrained file shows as follows:

Model at epoch 68: overall accuracy = 0.6127, mean_acc = 0.2182 epoch 68: 0th label: accuracy = 0.9814 epoch 68: 1th label: accuracy = 0.0 epoch 68: 2th label: accuracy = 0.0 epoch 68: 3th label: accuracy = 0.0 epoch 68: 4th label: accuracy = 0.0 epoch 68: 5th label: accuracy = 0.0 epoch 68: 6th label: accuracy = 0.0 epoch 68: 9th label: accuracy = 0.0 epoch 68: 10th label: accuracy = 0.9828

can you share any idea with the problem??

exogeny commented 4 years ago

And also, i think the code of balanced entropy loss is different from the article.