# TextZoom
[https://github.com/WenjiaWang0312/TextZoom](https://github.com/WenjiaWang0312/TextZoom)

## How to Use TextZoom
實驗環境: python 3.9, cuda 12.2
1. 進入TextZoom資料夾: `cd TextZoom`
2. 建立conda環境: `conda create -f environment.yml`，並且啟動環境:`conda activate TextZoom2`
3. 安裝一些套件(用環境檔安裝的時候沒辦法裝起來): `pip install -r requirements.txt`
4. 安裝pytorch: `conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia`
5. 將測資放在`src/documents/`資料夾下
6. 進入src資料夾: `cd src`
7. 執行程式碼:
`
python3 main.py --demo --demo_dir='./documents/' --resume='./ckpt/vis/pretrained.pth' --STN --mask --CLAHE --binarize
`
一些引數的意義如下：
* -\-demo_dir: 測資圖片資料夾
* -\-resume: pretrained weight的位置
* -\-CLAHE: 在做完SR之後是否要用CLAHE來自動調整亮度以及對比度
* -\-binarize: 做完SR之後是否要對圖片做二極化

8. SR結果將會出現在`src/documents/SR/`資料夾中