diff --git a/README.md b/README.md index 85a8d6811fdf05a8a34d38aa361a925af086f58b..780e6b8648ff30f87348997b67b4798966e6f060 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 -[Efficientb4](cv/classification/efficientb4/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 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 91% rename from cv/classification/efficientb4/pytorch/README.md rename to cv/classification/efficientnet_b4/pytorch/README.md index 7ac065b8f620ca9a7c25eb340fc6e0774f74bba3..aa4c4c9cf57c7b1058d08b337fcae6be32cfa891 100755 --- a/cv/classification/efficientb4/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 + + 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