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