thuml / CoTuning

Code release for NeurIPS 2020 paper "Co-Tuning for Transfer Learning"
39 stars 4 forks source link

CoTuning execution result problem #1

Closed sksksk1748 closed 3 years ago

sksksk1748 commented 3 years ago

My environment :

CUDA 10.1
python 3.7.10
pytorch == 1.7.1
torchvision == 0.8.2
torchaudio == 0.7.2
scikit-learn == 0.24.1
numpy == 1.20.1
argparse == 1.4.0 
tqdm == 4.59.0

Dataset Downloadlink : https://drive.google.com/file/d/1hbzc_P1FuxMkcabkgn9ZKinBwW683j45/view

Offical CUB-200-2011 dataset ( 001. ~ 200. ) image

My CUB-200-2011 dataset directory architecture :

CoTung
├── data/
│   └── finetune/ 
│       └── cub200/ 
│           ├── test/
│           │   └── test/
│           │       └── ... (151. ~ 200. image folder)
│           ├── train/
│           │   └── train/
│           │       └── ... (001. ~ 120. image folder)
│           └──  val/
│               └── val/
│                   └── ... (121. ~ 150. image folder)
├── module/
├── train.py
└── ...

My Quick Start :

$ python train.py --gpu 0 --data_path ./data/finetune/cub200/ --class_num 200 --batch_size 32

Result : image

How to solve ? Thanks !

Lanxiaozhi commented 3 years ago

@sksksk1748 I noticed that your directory architecture is not correct, it should be like this (just remove the redundant layers):

CoTung
├── data/
│   └── finetune/ 
│       └── cub200/ 
│           ├── test/
│           │       └── ... (about 200 folders)
│           ├── train/
│           │       └── ... (about 200 folders)
│           └──  val/
│                   └── ... (about 200 folders)
├── module/
├── train.py
└── ...

The split of the dataset should be based on all data rather than category (each folder should contain about 200 categories).

youkaichao commented 3 years ago

@sksksk1748 Where do you get the "official" split?

sksksk1748 commented 3 years ago

@Lanxiaozhi Great ! @youkaichao After I downloaded the dataset from the official, I divided the dataset by myself To be honest, this is not a wise method, because it is difficult to restore the results of your experiment. I hope you can provide information on how the datasets of CUB-200-2011, Stanford Cars, and FGVC Aircraft are divided on the path. By the way, after downloading the COCO70 dataset, the images in it are displayed abnormally, is it correct? In my opinion, should be able to see what this picture is.

youkaichao commented 3 years ago

@sksksk1748 what do you mean by "after downloading the COCO70 dataset, the images in it are displayed abnormally"? Can you provide some showcases? In my server, the images are displayed normally :)

youkaichao commented 3 years ago

The problem of "directory architecture" has been resolved and so this issue will be closed. But you are free to continue the discussion on COCO70 and we are glad to help you.