search

bermanmaxim / LovaszSoftmax

Code for the Lovász-Softmax loss (CVPR 2018)
http://bmax.im/LovaszSoftmax
MIT License
1.38k stars 269 forks source link

ENet version? #3

Open sourabhyadav opened 6 years ago

sourabhyadav commented 6 years ago

Dear Maxim, Nice to read your paper. Congrats for getting selected in the CVPR 18. Great joob!!! I am interested in fine tuned ENet version of your work. If you can provide relevant prototxt files and pre trained weights and/or modified caffe build would help me alot. Thanks

bermanmaxim commented 6 years ago

Hi @sourabhyadav, we will provide the trained weights later on. Keep in mind though that it won't be in caffe, I've been using my "ENet pytorch" implementation as described in https://github.com/bermanmaxim/Enet-PyTorch (this model is directly converted from the lua-torch model provided by the authors).