search

camicroscope / tfjs-models

Sample Models for use with caMicroscope
7 stars 4 forks source link

Adding IRRCNN model #1

Closed leoarc closed 4 years ago

leoarc commented 4 years ago

Adds another classification model.

Most of the code is from the already present IRCNN model . This one contains residual connections which adds the inputs to the outputs of each IRCNN block. It is based on the Inception_V4 instead of the Inception_V3 .

This is more of a version update as :

The parameters chosen are from one of the experiments done in the above paper.

The accuracy levels are not as good as in the paper as I only trained on 1/3rd of the data and for only 20 epochs (150 in the paper) .( I'll try to train it more afterwards .)

I have tested the tfjs model in caMicroscope and its working.

@Insiyaa Do you think this is worth adding ?

Insiyaa commented 4 years ago

This is a good addition. If feasible, do train this further! I already see the patch size in the readme, could you please add if the model is trained on RGB/b&w images and pre-processing used for those using the tfjs model directly?

leoarc commented 4 years ago

Done !

Insiyaa commented 4 years ago

Good! Please also add magnification.

leoarc commented 4 years ago

Sorry for not including everything at once . Anything else I've missed ?