search

FengyangZhang / caffe_roomnet

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

how to train？ #2

Open liu09114 opened 7 years ago

liu09114 commented 7 years ago

@FengyangZhang Thanks for your released code. I want to train a model for keypoints prediction, but I can't get the type of input data. Will you offer the code for preparing data? Or just release the type of 'train.txt' as in the train.prototxt?

FengyangZhang commented 7 years ago

Hi Liu, It's a comma-separated text file, I used some simple script to produce it and I don't have the script anymore. However, I think you can find out the format in src/caffe/layers/data_heatmap.cpp when the code reads the input data into memory.

liu09114 commented 7 years ago

Yes, I have read the layer code data_heatmap.cpp. Since now I just want to have a quick test on the keypoint detection. Could you provide your 'train.txt' to me? My email is liu09114@gmail.com . Best wishes.

FengyangZhang commented 7 years ago

I'm sorry but I don't have that now..