From 03d30e0018c6fcf0f961161be9e3091ec1924ce8 Mon Sep 17 00:00:00 2001
From: JunHyuk Jang <108822253+jjunhyukki@users.noreply.github.com>
Date: Mon, 10 Oct 2022 02:35:00 +0900
Subject: [PATCH] Update README.md

---
 README.md | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 90da484..43a51b4 100644
--- a/README.md
+++ b/README.md
@@ -45,4 +45,24 @@ streamlit run super_resolution.py
 
 2. modules
 * pretrained models : SwinIR pretrained models
-* 
+* dataset.py : Transform images and build custom dataset
+* edsr.py : Customized EDSR model
+* swinir.py : Customized SwinIR model
+* losses.py : For L1 loss in train process
+* model.py : Our own made super-resolution model code
+
+3. others
+* inference_dacon.py : Code of super resolution for test set
+* main.py : The whole process from building dataset to super resolution
+* train.py : Train process code
+
+### results
+* CustomModel_30.pt : pretrained custom super resolution model
+
+### streamlit
+* app_funcs.py : Functions that are used in streamlit app
+* super_resolution.py : Code to run streamlit app
+* uploads : test images
+
+### others
+* requirements.txt : Required dependencies to run the streamlit app