Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add image augmentation for csv dataloader #17

Merged
merged 1 commit into from
May 19, 2022

Conversation

TomonobuTsujikawa
Copy link
Contributor

Introduce normalize and augment images provided by csv dataset.
Here is the example of json.
Currently, augmentation type can be selected from imagenet and cifar10.

"dataloader": {
    "csv": {
        "train_portion": 0.96,
        "train_file": "ilsvrc-train.csv",
        "valid_file": "ilsvrc-test.csv",
        "train_cache_dir": "ilsvrc-train.cache",
        "valid_cache_dir": "ilsvrc-test.cache",
        "augmentation": {
            "type": "imagenet"
        }
    }
},

Also, input_shapes should be set actual input image size like [3, 320, 320] or [3, 480,480].

@TomonobuTsujikawa TomonobuTsujikawa added the release-note-utility Auto-release; Utilities label Apr 25, 2022
@YukioOobuchi YukioOobuchi merged commit 718bc2d into master May 19, 2022
@YukioOobuchi YukioOobuchi deleted the feature/20220329-fix-csv branch May 19, 2022 06:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
release-note-utility Auto-release; Utilities
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants