espejelomar
commited on
Commit
•
99a3944
1
Parent(s):
13cc45c
Update README.md
Browse files
README.md
CHANGED
@@ -22,9 +22,13 @@ Disclaimer: The model was fine-tuned after [Chapter 5](https://github.com/fastai
|
|
22 |
|
23 |
The model was finetuned using the `cnn_learner` method of the fastai library suing a Resnet 34 backbone pretrained on the ImageNet dataset. The fastai library uses PyTorch for the undelying operations. `cnn_learner` automatically gets a pretrained model from a given architecture with a custom head that is suitable for the target data.
|
24 |
|
25 |
-
Resnet34 is a 34 layer convolutional neural network. It takes residuals from each layer and uses them in the subsequent connected layers.
|
|
|
|
|
26 |
|
27 |
-
|
|
|
|
|
28 |
|
29 |
```
|
30 |
learn = cnn_learner(dls, resnet34, metrics=error_rate)
|
|
|
22 |
|
23 |
The model was finetuned using the `cnn_learner` method of the fastai library suing a Resnet 34 backbone pretrained on the ImageNet dataset. The fastai library uses PyTorch for the undelying operations. `cnn_learner` automatically gets a pretrained model from a given architecture with a custom head that is suitable for the target data.
|
24 |
|
25 |
+
Resnet34 is a 34 layer convolutional neural network. It takes residuals from each layer and uses them in the subsequent connected layers. Advantages of a resnet arquitecture ([Neurohive, 2019](https://neurohive.io/en/popular-networks/resnet/)):
|
26 |
+
- Are easy to optimize, but the “plain” networks (that simply stack layers) shows higher training error when the depth increases.
|
27 |
+
- Can easily gain accuracy from greatly increased depth, producing results which are better than previous networks.
|
28 |
|
29 |
+
Please refer to the original paper '[Deep Residual Learning for Image Recognition](https://arxiv.org/pdf/1512.03385.pdf)' written by Kaiming He, Xiangyu Zhang, Shaoqing Ren and Jian Sun.
|
30 |
+
|
31 |
+
Specifically, the model was obtained:
|
32 |
|
33 |
```
|
34 |
learn = cnn_learner(dls, resnet34, metrics=error_rate)
|