diff --git a/hydra_flash/conf/datamodule/movie_posters.yaml b/hydra_flash/conf/datamodule/movie_posters.yaml new file mode 100644 index 0000000000000000000000000000000000000000..a2cfa322dc7c6fc56ec1cf4103cc874d28c3ae5f --- /dev/null +++ b/hydra_flash/conf/datamodule/movie_posters.yaml @@ -0,0 +1,14 @@ +_target_: flash.image.ImageClassificationData.from_csv +input_field: "Id" +target_fields: ["Action", "Romance", "Crime", "Thriller", "Adventure"] +train_file: ${preprocess.path}/movie_posters/train/metadata.csv +train_resolver: + _target_: hydra_flash.utils.resolver + _partial_: true +val_file: ${preprocess.path}/movie_posters/val/metadata.csv +val_resolver: + _target_: hydra_flash.utils.resolver + _partial_: true +transform_kwargs: + image_size: [128, 128] +batch_size: 1 diff --git a/hydra_flash/conf/datamodule/movie_posters_test.yaml b/hydra_flash/conf/datamodule/movie_posters_test.yaml new file mode 100644 index 0000000000000000000000000000000000000000..9ad4083e560898b7224d1bd742fe500a4fc7c531 --- /dev/null +++ b/hydra_flash/conf/datamodule/movie_posters_test.yaml @@ -0,0 +1,6 @@ +_target_: flash.image.ImageClassificationData.from_files +predict_files: + - "${preprocess.path}/movie_posters/predict/tt0085318.jpg" + - "${preprocess.path}/movie_posters/predict/tt0089461.jpg" + - "${preprocess.path}/movie_posters/predict/tt0097179.jpg" +batch_size: 3 diff --git a/hydra_flash/conf/experiment/movie_poster.yaml b/hydra_flash/conf/experiment/movie_poster.yaml new file mode 100644 index 0000000000000000000000000000000000000000..ba553527228eae64a8f9f82553d7638de5e2019c --- /dev/null +++ b/hydra_flash/conf/experiment/movie_poster.yaml @@ -0,0 +1,13 @@ +# @package _global_ +defaults: + - _self_ + - /server: local + - /preprocess: movie_poster + - /datamodule: movie_posters + - /model: resnet18 + - /datamodule@predict_datamodule: movie_posters_test + +trainer: + _target_: flash.Trainer + max_epochs: 3 + gpus: 1 diff --git a/hydra_flash/conf/preprocess/movie_poster.yaml b/hydra_flash/conf/preprocess/movie_poster.yaml new file mode 100644 index 0000000000000000000000000000000000000000..492810af10f7a622af9bee7599073a17be8e184b --- /dev/null +++ b/hydra_flash/conf/preprocess/movie_poster.yaml @@ -0,0 +1,4 @@ +defaults: + - download + +url: "https://pl-flash-data.s3.amazonaws.com/movie_posters.zip"