From e557f2a08039050e4a1f35874474c570908dc225 Mon Sep 17 00:00:00 2001 From: Victor Joos <victor.joos@uclouvain.be> Date: Tue, 28 Jun 2022 15:03:09 +0200 Subject: [PATCH] Other task --- hydra_flash/conf/datamodule/movie_posters.yaml | 14 ++++++++++++++ .../conf/datamodule/movie_posters_test.yaml | 6 ++++++ hydra_flash/conf/experiment/movie_poster.yaml | 13 +++++++++++++ hydra_flash/conf/preprocess/movie_poster.yaml | 4 ++++ 4 files changed, 37 insertions(+) create mode 100644 hydra_flash/conf/datamodule/movie_posters.yaml create mode 100644 hydra_flash/conf/datamodule/movie_posters_test.yaml create mode 100644 hydra_flash/conf/experiment/movie_poster.yaml create mode 100644 hydra_flash/conf/preprocess/movie_poster.yaml diff --git a/hydra_flash/conf/datamodule/movie_posters.yaml b/hydra_flash/conf/datamodule/movie_posters.yaml new file mode 100644 index 0000000..a2cfa32 --- /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 0000000..9ad4083 --- /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 0000000..ba55352 --- /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 0000000..492810a --- /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" -- GitLab