dhlab-epfl / dhSegment

Generic framework for historical document processing
https://dhlab-epfl.github.com/dhSegment
GNU General Public License v3.0
370 stars 116 forks source link

A more efficient neural architecture #39

Closed mrocr closed 5 years ago

mrocr commented 5 years ago

@solivr @SeguinBe Thank you for your hard work,

Can you merge Mobilenet v2 with master, along with adding a demo for using it. Thank you

Waiting for your reply

mrocr commented 5 years ago

@SeguinBe have a look at this improved Mobilenet v2, which you can download from here. It can achieve 70% better FLOPs.

SeguinBe commented 5 years ago

The mobileNet experiment was more about trying to add a new network and check the flexibility of a new form of the framework.

While I tried training, and it was running. In my recollection the performance was not on par with the bigger ones for my test case.

I skimmed over the link you provided and it is mainly about model compression. A compressed model can not be used for training right away. What you want is compressing the final trained model of dhSegment, and this is out of the scope of this project.

mrocr commented 5 years ago

@SeguinBe @solivr The goal is to find a neural network that is capable of delivering Resnet-50's accuracy, without being over-parameterized or heavy. Thus, the idea of finding that secret sauce, that is able to run with speed & accuracy even when used on a CPU is actually possible, and have been already achieved.

Have a look at: