search

FengyangZhang / caffe_roomnet

Segnet-like CNN for room layout estimation
Other
27 stars 7 forks source link

Weights #1

Open reynoldscem opened 7 years ago

reynoldscem commented 7 years ago

Hello,

I was wondering if you intended to release weights for this model?

FengyangZhang commented 7 years ago

Hi,

Actually you can train the model now. I've updated the README.md.

reynoldscem commented 7 years ago

I am presuming you have successfully trained a model on LSUN already?

FengyangZhang commented 7 years ago

Yes.

reynoldscem commented 7 years ago

That's excellent. What sort of performance are you getting compared to the authors?

FengyangZhang commented 7 years ago

Pretty close I think, though the classification head is not producing good enough predictions.

reynoldscem commented 7 years ago

That's very good to hear. Do you have scripts which you used to prepare your data?

FengyangZhang commented 7 years ago

Sorry but I don't have a very intact piece of code of preparing training data... It was some matlab plus some awk.