From 0844714500f0eae812315d33b39556d203eb4ca2 Mon Sep 17 00:00:00 2001 From: majorli6 Date: Mon, 21 Nov 2022 11:32:46 +0800 Subject: [PATCH 1/2] ssd prepare.py bugfix link #I5Y00S Signed-off-by: majorli6 --- cv/classification/wavemlp/pytorch/README.md | 4 ++-- cv/detection/ssd/pytorch/base/prepare.py | 4 ++-- cv/detection/ssd/pytorch/iluvatar/csrc/nhwc/Descriptors.cpp | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/cv/classification/wavemlp/pytorch/README.md b/cv/classification/wavemlp/pytorch/README.md index a8b30f44d..7275eb6ed 100644 --- a/cv/classification/wavemlp/pytorch/README.md +++ b/cv/classification/wavemlp/pytorch/README.md @@ -6,7 +6,7 @@ In the field of computer vision, recent works show that a pure MLP architecture ## Step 1: Installing packages ``` -pip install thop timm==0.3.2 torchprofile +pip install thop timm==0.4.5 torchprofile ``` ## Step 2: Training @@ -40,4 +40,4 @@ bash run.sh /home/datasets/cv/ImageNet/ ## Reference -https://github.com/huawei-noah/Efficient-AI-Backbones/blob/master/wavemlp_pytorch/ \ No newline at end of file +https://github.com/huawei-noah/Efficient-AI-Backbones/blob/master/wavemlp_pytorch/ diff --git a/cv/detection/ssd/pytorch/base/prepare.py b/cv/detection/ssd/pytorch/base/prepare.py index 0a9eefd5f..b85fb7522 100644 --- a/cv/detection/ssd/pytorch/base/prepare.py +++ b/cv/detection/ssd/pytorch/base/prepare.py @@ -150,9 +150,9 @@ def prepare_data(): if exist_preprocessed_data: return - current_dir = os.path.join(MODEL_DIR, "pytorch") + # current_dir = os.path.join(MODEL_DIR, "pytorch") cmds = [ - f"cd {current_dir}", + # f"cd {current_dir}", f"python3 data_preprocessing/prepare_json.py --keep-keys {DATA_DIR}/annotations/instances_val2017.json {DATA_DIR}/annotations/bbox_only_instances_val2017.json", f"python3 data_preprocessing/prepare_json.py {DATA_DIR}/annotations/instances_train2017.json {DATA_DIR}/annotations/bbox_only_instances_train2017.json" ] diff --git a/cv/detection/ssd/pytorch/iluvatar/csrc/nhwc/Descriptors.cpp b/cv/detection/ssd/pytorch/iluvatar/csrc/nhwc/Descriptors.cpp index f3143f923..afe0466c9 100644 --- a/cv/detection/ssd/pytorch/iluvatar/csrc/nhwc/Descriptors.cpp +++ b/cv/detection/ssd/pytorch/iluvatar/csrc/nhwc/Descriptors.cpp @@ -20,7 +20,7 @@ #include "Descriptors.h" #include - +#include #include #include #include -- Gitee From 35e3f47a17a2a51f8a7c21b0c24a27b0c846074f Mon Sep 17 00:00:00 2001 From: majorli6 Date: Mon, 21 Nov 2022 13:52:05 +0800 Subject: [PATCH 2/2] trimm version to 0.4.5 link #I5Y00S Signed-off-by: majorli6 --- cv/classification/convnext/pytorch/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cv/classification/convnext/pytorch/README.md b/cv/classification/convnext/pytorch/README.md index f193f7a4f..de160e208 100644 --- a/cv/classification/convnext/pytorch/README.md +++ b/cv/classification/convnext/pytorch/README.md @@ -5,7 +5,7 @@ The ConvNeXT model was proposed in [A ConvNet for the 2020s](https://arxiv.org/a ## Step 1: Installing ```bash -pip install timm==0.3.2 tensorboardX six torch torchvision +pip install timm==0.4.5 tensorboardX six torch torchvision ``` Sign up and login in [imagenet official website](https://www.image-net.org/index.php), then choose 'Download' to download the whole imagenet dataset. Specify `/path/to/imagenet` to your imagenet path in later training process. -- Gitee