Giters

BMIRDS / deepslide

Code for the Nature Scientific Reports paper "Pathologist-level classification of histologic patterns on resected lung adenocarcinoma slides with deep neural networks." A sliding window framework for classification of high resolution whole-slide images, often microscopy or histopathology images.

Home Page:https://www.nature.com/articles/s41598-019-40041-7

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Use pre-trained ResNet with frozen layers

Tato14 opened this issue · comments

commented

Hi!
As transfer learning seem to be quite a big issue on these days, I was wondering if you tried to train ResNet 18 using ImageNet pretrained model but freezing all the layers except the last one.

Also, I found the documentenation in PyTorch here but I am not sure where should I implement this lines of code. Should be in the create_model function?
Thanks!

commented

Also, it would be great if you could share the weights of your ResNet 18 (and 34?) to give them a try as a pretrained model.

commented

You can set pretrain=True in code/config.py to instantiate the ImageNet pretrained model. Currently, the weights are not available, but they will be posted in this repo once they are.