From 23d5fa15ad301adb301455bfc1f2e917823059b7 Mon Sep 17 00:00:00 2001 From: majorli Date: Mon, 24 Jun 2024 14:12:30 +0800 Subject: [PATCH 1/3] bugfix: efficientnet b4 path not correct Signed-off-by: majorli --- .../{efficientb4 => efficientnet_b4}/pytorch/.gitignore | 0 .../{efficientb4 => efficientnet_b4}/pytorch/README.md | 0 .../{efficientb4 => efficientnet_b4}/pytorch/__init__.py | 0 .../pytorch/common_utils/__init__.py | 0 .../{efficientb4 => efficientnet_b4}/pytorch/common_utils/dist.py | 0 .../{efficientb4 => efficientnet_b4}/pytorch/common_utils/loss.py | 0 .../pytorch/common_utils/metric_logger.py | 0 .../{efficientb4 => efficientnet_b4}/pytorch/common_utils/misc.py | 0 .../pytorch/common_utils/smooth_value.py | 0 .../pytorch/dataloader/__init__.py | 0 .../pytorch/dataloader/classification.py | 0 .../pytorch/dataloader/dali_classification.py | 0 .../pytorch/dataloader/utils/__init__.py | 0 .../pytorch/dataloader/utils/presets_classification.py | 0 .../{efficientb4 => efficientnet_b4}/pytorch/train.py | 0 .../{efficientb4 => efficientnet_b4}/pytorch/utils.py | 0 16 files changed, 0 insertions(+), 0 deletions(-) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/.gitignore (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/README.md (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/__init__.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/__init__.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/dist.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/loss.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/metric_logger.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/misc.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/common_utils/smooth_value.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/dataloader/__init__.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/dataloader/classification.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/dataloader/dali_classification.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/dataloader/utils/__init__.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/dataloader/utils/presets_classification.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/train.py (100%) rename cv/classification/{efficientb4 => efficientnet_b4}/pytorch/utils.py (100%) diff --git a/cv/classification/efficientb4/pytorch/.gitignore b/cv/classification/efficientnet_b4/pytorch/.gitignore similarity index 100% rename from cv/classification/efficientb4/pytorch/.gitignore rename to cv/classification/efficientnet_b4/pytorch/.gitignore diff --git a/cv/classification/efficientb4/pytorch/README.md b/cv/classification/efficientnet_b4/pytorch/README.md similarity index 100% rename from cv/classification/efficientb4/pytorch/README.md rename to cv/classification/efficientnet_b4/pytorch/README.md diff --git a/cv/classification/efficientb4/pytorch/__init__.py b/cv/classification/efficientnet_b4/pytorch/__init__.py similarity index 100% rename from cv/classification/efficientb4/pytorch/__init__.py rename to cv/classification/efficientnet_b4/pytorch/__init__.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/__init__.py b/cv/classification/efficientnet_b4/pytorch/common_utils/__init__.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/__init__.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/__init__.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/dist.py b/cv/classification/efficientnet_b4/pytorch/common_utils/dist.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/dist.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/dist.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/loss.py b/cv/classification/efficientnet_b4/pytorch/common_utils/loss.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/loss.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/loss.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/metric_logger.py b/cv/classification/efficientnet_b4/pytorch/common_utils/metric_logger.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/metric_logger.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/metric_logger.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/misc.py b/cv/classification/efficientnet_b4/pytorch/common_utils/misc.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/misc.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/misc.py diff --git a/cv/classification/efficientb4/pytorch/common_utils/smooth_value.py b/cv/classification/efficientnet_b4/pytorch/common_utils/smooth_value.py similarity index 100% rename from cv/classification/efficientb4/pytorch/common_utils/smooth_value.py rename to cv/classification/efficientnet_b4/pytorch/common_utils/smooth_value.py diff --git a/cv/classification/efficientb4/pytorch/dataloader/__init__.py b/cv/classification/efficientnet_b4/pytorch/dataloader/__init__.py similarity index 100% rename from cv/classification/efficientb4/pytorch/dataloader/__init__.py rename to cv/classification/efficientnet_b4/pytorch/dataloader/__init__.py diff --git a/cv/classification/efficientb4/pytorch/dataloader/classification.py b/cv/classification/efficientnet_b4/pytorch/dataloader/classification.py similarity index 100% rename from cv/classification/efficientb4/pytorch/dataloader/classification.py rename to cv/classification/efficientnet_b4/pytorch/dataloader/classification.py diff --git a/cv/classification/efficientb4/pytorch/dataloader/dali_classification.py b/cv/classification/efficientnet_b4/pytorch/dataloader/dali_classification.py similarity index 100% rename from cv/classification/efficientb4/pytorch/dataloader/dali_classification.py rename to cv/classification/efficientnet_b4/pytorch/dataloader/dali_classification.py diff --git a/cv/classification/efficientb4/pytorch/dataloader/utils/__init__.py b/cv/classification/efficientnet_b4/pytorch/dataloader/utils/__init__.py similarity index 100% rename from cv/classification/efficientb4/pytorch/dataloader/utils/__init__.py rename to cv/classification/efficientnet_b4/pytorch/dataloader/utils/__init__.py diff --git a/cv/classification/efficientb4/pytorch/dataloader/utils/presets_classification.py b/cv/classification/efficientnet_b4/pytorch/dataloader/utils/presets_classification.py similarity index 100% rename from cv/classification/efficientb4/pytorch/dataloader/utils/presets_classification.py rename to cv/classification/efficientnet_b4/pytorch/dataloader/utils/presets_classification.py diff --git a/cv/classification/efficientb4/pytorch/train.py b/cv/classification/efficientnet_b4/pytorch/train.py similarity index 100% rename from cv/classification/efficientb4/pytorch/train.py rename to cv/classification/efficientnet_b4/pytorch/train.py diff --git a/cv/classification/efficientb4/pytorch/utils.py b/cv/classification/efficientnet_b4/pytorch/utils.py similarity index 100% rename from cv/classification/efficientb4/pytorch/utils.py rename to cv/classification/efficientnet_b4/pytorch/utils.py -- Gitee From bc7aa31d58c23e4aa26f87c482f1494ddda2add5 Mon Sep 17 00:00:00 2001 From: majorli Date: Mon, 24 Jun 2024 14:14:30 +0800 Subject: [PATCH 2/3] bugfix: efficientnet b4 model name not correct Signed-off-by: majorli --- README.md | 2 +- .../efficientnet_b4/pytorch/README.md | 14 +++++++++++--- 2 files changed, 12 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 85a8d6811..a67c9f732 100644 --- a/README.md +++ b/README.md @@ -80,7 +80,7 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 [DPN107](cv/classification/dpn107/pytorch/README.md) | PyTorch | ImageNet [ECA-MobileNetV2](cv/classification/eca_mobilenet_v2/pytorch/README.md) | PyTorch | ImageNet [ECA-ResNet152](cv/classification/eca_resnet152/pytorch/README.md) | PyTorch | ImageNet -[Efficientb4](cv/classification/efficientb4/pytorch/README.md) | PyTorch | ImageNet +[EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet [EfficientNetB0](cv/classification/efficientnet_b0/paddlepaddle/README.md) | PaddlePaddle | ImageNet [FasterNet](cv/classification/fasternet/pytorch/README.md)  | PyTorch | ImageNet [GoogLeNet](cv/classification/googlenet/pytorch/README.md) | PyTorch | ImageNet diff --git a/cv/classification/efficientnet_b4/pytorch/README.md b/cv/classification/efficientnet_b4/pytorch/README.md index 7ac065b8f..aa4c4c9cf 100755 --- a/cv/classification/efficientnet_b4/pytorch/README.md +++ b/cv/classification/efficientnet_b4/pytorch/README.md @@ -1,7 +1,11 @@ -# efficientb4 +# EfficientNetB4 + ## Model description + EfficientNet is a convolutional neural network architecture and scaling method that uniformly scales all dimensions of depth/width/resolution using a compound coefficient. + ## Step 1: Installing + ```bash pip3 install torch torchvision ``` @@ -24,16 +28,20 @@ imagenet └── val_list.txt ``` - ## Step 2: Training + ### One single GPU + ```bash python3 train.py --data-path /path/to/imagenet --model efficientnet_b4 --batch-size 128 ``` + ### Multiple GPUs on one machine + ```bash python3 -m torch.distributed.launch --nproc_per_node=8 --use_env train.py --data-path /path/to/imagenet --model efficientnet_b4 --batch-size 128 ``` ## Reference -https://github.com/pytorch/vision/blob/main/torchvision/models/efficientnet.py + + -- Gitee From 598c88f983cc490b8928b1b1ca9972746c24f1ae Mon Sep 17 00:00:00 2001 From: majorli6 Date: Mon, 24 Jun 2024 06:18:06 +0000 Subject: [PATCH 3/3] update README.md - reorder Signed-off-by: majorli6 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index a67c9f732..780e6b864 100644 --- a/README.md +++ b/README.md @@ -80,8 +80,8 @@ DeepSparkHub甄选上百个应用算法和模型，覆盖AI和通用计算各领 [DPN107](cv/classification/dpn107/pytorch/README.md) | PyTorch | ImageNet [ECA-MobileNetV2](cv/classification/eca_mobilenet_v2/pytorch/README.md) | PyTorch | ImageNet [ECA-ResNet152](cv/classification/eca_resnet152/pytorch/README.md) | PyTorch | ImageNet -[EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet [EfficientNetB0](cv/classification/efficientnet_b0/paddlepaddle/README.md) | PaddlePaddle | ImageNet +[EfficientNetB4](cv/classification/efficientnet_b4/pytorch/README.md) | PyTorch | ImageNet [FasterNet](cv/classification/fasternet/pytorch/README.md)  | PyTorch | ImageNet [GoogLeNet](cv/classification/googlenet/pytorch/README.md) | PyTorch | ImageNet [GoogLeNet](cv/classification/googlenet/paddlepaddle/README.md) | PaddlePaddle | ImageNet -- Gitee