wentaozhu / deep-mil-for-whole-mammogram-classification

Zhu, Wentao, Qi Lou, Yeeleng Scott Vang, and Xiaohui Xie. "Deep Multi-instance Networks with Sparse Label Assignment for Whole Mammogram Classification." MICCAI 2017.
MIT License
113 stars 36 forks source link

which is the backend for keras, theano or tensorflow? #9

Closed lxmwust closed 3 years ago

lxmwust commented 6 years ago

att, it seems to be using theano. I tried to use tensorflow, it report a dimension error for Alexnet network. I would be better to declare it clearly in the document. Besides, the modified convnetskeras (outdated) also reported several errors. The code is not easy to setup.

wentaozhu commented 6 years ago

Yes. It is theano backend. I think now keras has added many features and there is no one to support theano any more. I am very sorry for the outdated. It is very hard to maintenance a project for a long time. But I think it is easy for you to update it based on the existing code because there are some api function interfaces changed.

lxmwust commented 6 years ago

Thank you for the reply. While I still think the update is not so easy. In fact, I tried to adapt the code. One problem I encountered is that it depends "convnets-keras" (which is theano based, and not supported anymore). I tried to modify convnets-keras to use tensorflow, after went through "issues" regard it and made some small changes, while I can run convnets-keras now, but the result I obtained is much worse than the result show on 'convnets-keras'. Thus I guess I will not be able run your code, since I am not familar to theano, and can not solve the 'convnets-keras' depence problem.

Estarcontigo commented 6 years ago

@lxmwust Hello, I want to know if you solved this problem? What is the solution? Is the code running? Looking forward to your reply.

LLYXC commented 6 years ago

Modify your Keras config ~/.keras/keras.json, set backend tensorflow, image_data_format to channel_first will do. Still some other changes in the custom_layers should be made in order to run the code.