From 72b54487222d56cde9e4e7237428114151b81e78 Mon Sep 17 00:00:00 2001 From: Shikun Li Date: Mon, 14 Nov 2022 14:17:04 +0800 Subject: [PATCH] Add how to obtain dataset --- cv/classification/acmix/pytorch/README.md | 1 + cv/classification/convnext/pytorch/README.md | 3 +++ cv/detection/retinanet/pytorch/README.md | 2 ++ 3 files changed, 6 insertions(+) diff --git a/cv/classification/acmix/pytorch/README.md b/cv/classification/acmix/pytorch/README.md index 6836fc1af..b7f5e7244 100644 --- a/cv/classification/acmix/pytorch/README.md +++ b/cv/classification/acmix/pytorch/README.md @@ -9,6 +9,7 @@ Convolution and self-attention are two powerful techniques for representation le pip install termcolor==1.1.0 yacs==0.1.8 timm==0.4.5 ``` +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. ## Step 2: Training diff --git a/cv/classification/convnext/pytorch/README.md b/cv/classification/convnext/pytorch/README.md index 16e6232d5..f193f7a4f 100644 --- a/cv/classification/convnext/pytorch/README.md +++ b/cv/classification/convnext/pytorch/README.md @@ -8,6 +8,9 @@ The ConvNeXT model was proposed in [A ConvNet for the 2020s](https://arxiv.org/a pip install timm==0.3.2 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. + + ## Step 2: Training ### Multiple GPUs on one machine ```bash diff --git a/cv/detection/retinanet/pytorch/README.md b/cv/detection/retinanet/pytorch/README.md index 0512a21a3..e4c5838ef 100644 --- a/cv/detection/retinanet/pytorch/README.md +++ b/cv/detection/retinanet/pytorch/README.md @@ -14,6 +14,8 @@ pip3 install 'scipy' 'matplotlib' 'pycocotools' 'opencv-python' 'easydict' 'tqdm ## Step 2: Training on COCO dataset +Download the [COCO Dataset](https://cocodataset.org/#home) + ### Multiple GPUs on one machine ```shell -- Gitee