Merge branch 'dev' of https://github.com/WZMIAOMIAO/deep-learning-for…

…-image-processing into dev
Lycodeboy · Feb 9, 2021 · 1a5cff5 · 1a5cff5
2 parents 76992ee + e5fb78e
commit 1a5cff5
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 7 deletions.
diff --git a/README.md b/README.md
@@ -32,7 +32,9 @@
 
   * ResNet（已完成）
     * [ResNet理论讲解](https://www.bilibili.com/video/BV1T7411T7wa)
+    * [ResNeXt理论讲解](https://www.bilibili.com/video/BV1Ap4y1p71v/)
     * [Pytorch搭建ResNet网络](https://www.bilibili.com/video/BV14E411H7Uw)
+    * [Pytorch搭建ResNeXt网络](https://www.bilibili.com/video/BV1rX4y1N7tE)
     * [Tensorflow2搭建ResNet网络](https://www.bilibili.com/video/BV1WE41177Ya)
 
   * MobileNet（已完成）

diff --git a/pytorch_classification/Test2_alexnet/train.py b/pytorch_classification/Test2_alexnet/train.py
@@ -55,7 +55,7 @@ def main():
                                                   batch_size=4, shuffle=False,
                                                   num_workers=nw)
 
-    print("using {} images for training, {} images fot validation.".format(train_num,
+    print("using {} images for training, {} images for validation.".format(train_num,
                                                                            val_num))
     # test_data_iter = iter(validate_loader)
     # test_image, test_label = test_data_iter.next()

diff --git a/pytorch_classification/Test3_vggnet/train.py b/pytorch_classification/Test3_vggnet/train.py
@@ -52,7 +52,7 @@ def main():
     validate_loader = torch.utils.data.DataLoader(validate_dataset,
                                                   batch_size=batch_size, shuffle=False,
                                                   num_workers=nw)
-    print("using {} images for training, {} images fot validation.".format(train_num,
+    print("using {} images for training, {} images for validation.".format(train_num,
                                                                            val_num))
 
     # test_data_iter = iter(validate_loader)

diff --git a/pytorch_classification/Test4_googlenet/train.py b/pytorch_classification/Test4_googlenet/train.py
@@ -53,7 +53,7 @@ def main():
                                                   batch_size=batch_size, shuffle=False,
                                                   num_workers=nw)
 
-    print("using {} images for training, {} images fot validation.".format(train_num,
+    print("using {} images for training, {} images for validation.".format(train_num,
                                                                            val_num))
 
     # test_data_iter = iter(validate_loader)

diff --git a/pytorch_classification/Test5_resnet/model.py b/pytorch_classification/Test5_resnet/model.py
@@ -170,7 +170,7 @@ def resnet101(num_classes=1000, include_top=True):
     return ResNet(Bottleneck, [3, 4, 23, 3], num_classes=num_classes, include_top=include_top)
 
 
-def resnet50_32x4d(num_classes=1000, include_top=True):
+def resnext50_32x4d(num_classes=1000, include_top=True):
     # https://download.pytorch.org/models/resnext50_32x4d-7cdf4587.pth
     groups = 32
     width_per_group = 4
@@ -181,7 +181,7 @@ def resnet50_32x4d(num_classes=1000, include_top=True):
                   width_per_group=width_per_group)
 
 
-def resnet101_32x8d(num_classes=1000, include_top=True):
+def resnext101_32x8d(num_classes=1000, include_top=True):
     # https://download.pytorch.org/models/resnext101_32x8d-8ba56ff5.pth
     groups = 32
     width_per_group = 8

diff --git a/pytorch_classification/Test5_resnet/train.py b/pytorch_classification/Test5_resnet/train.py
@@ -54,7 +54,7 @@ def main():
                                                   batch_size=batch_size, shuffle=False,
                                                   num_workers=nw)
 
-    print("using {} images for training, {} images fot validation.".format(train_num,
+    print("using {} images for training, {} images for validation.".format(train_num,
                                                                            val_num))
 
     net = resnet34()

diff --git a/pytorch_classification/Test6_mobilenet/train.py b/pytorch_classification/Test6_mobilenet/train.py
@@ -56,7 +56,7 @@ def main():
                                                   batch_size=batch_size, shuffle=False,
                                                   num_workers=nw)
 
-    print("using {} images for training, {} images fot validation.".format(train_num,
+    print("using {} images for training, {} images for validation.".format(train_num,
                                                                            val_num))
 
     # create model