albumentations-team / autoalbument-benchmarks

Benchmarks for AutoAlbument - AutoML for Image Augmentation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to run on custom dataset?

huyhoangle86 opened this issue · comments

Hi, thank for your works.
As the title said, I have my own dataset, how can I use your code ?

@huyhoangle86. To the best of my understanding, you can use autoalbument library: https://albumentations.ai/docs/autoalbument/installation/ and follow the documents
https://albumentations.ai/docs/autoalbument/how_to_use/ -> autoalbument-create -> Configure custom task
https://albumentations.ai/docs/autoalbument/examples/list/ -> autoalbument-search -> Run Faster Autoaugment

@huyhoangle86 Have you been able to replicate their result under the baseline method yet? Many thanks.

Augmentation strategy | Top-1 Accuracy | Top-5 Accuracy -- | -- | -- Baseline | 91.79 | 99.63