61 lines
1.7 KiB
Markdown
61 lines
1.7 KiB
Markdown
|
unitprint/backend/scan/DocRes
|
||
|
|
=
|
||
|
|
https://github.com/ZZZHANG-jx/DocRes.git
|
||
|
|
|
||
|
|
``` sh *** build ***
|
||
|
|
mv * ..
|
||
|
|
```
|
||
|
|
|
||
|
|
``` sh *** config ***
|
||
|
|
ln -sf /pkg/gnu/unitprint/backend/scan/venv
|
||
|
|
source venv/bin/activate
|
||
|
|
pip install einops scikit-image opencv-python-headless tqdm
|
||
|
|
```
|
||
|
|
|
||
|
|
``` sh *** upload_weights ***
|
||
|
|
mkdir -p data/weights
|
||
|
|
#rsync -avz ~/Downloads/mbd.pkl a:/pkg/gnu/unitprint/backend/scan/DocRes/data/weights
|
||
|
|
#rsync -avz ~/Downloads/docres.pkl a:/pkg/gnu/unitprint/backend/scan/DocRes/data/weights
|
||
|
|
```
|
||
|
|
|
||
|
|
mkdir -p ~/.config/miopen
|
||
|
|
export MIOPEN_USER_DB_PATH="/root/.config/miopen"
|
||
|
|
export MIOPEN_FIND_MODE=3
|
||
|
|
export HIP_VISIBLE_DEVICES=0
|
||
|
|
|
||
|
|
time ../../venv/bin/python inference.py \
|
||
|
|
--task deshadowing \
|
||
|
|
--im_path ./IMG_20260905_172410_rec.jpg \
|
||
|
|
--out_folder ./
|
||
|
|
|
||
|
|
time python inference.py \
|
||
|
|
--task appearance \
|
||
|
|
--im_path ./output_pipeline/IMG_20260905_172410_rec_deshadowing.png \
|
||
|
|
--out_folder ./output_pipeline
|
||
|
|
|
||
|
|
time python inference.py \
|
||
|
|
--task deblurring \
|
||
|
|
--im_path ./output_pipeline/IMG_20260905_172410_rec_deshadowing_appearance.png \
|
||
|
|
--out_folder ./output_pipeline
|
||
|
|
|
||
|
|
time python inference.py \
|
||
|
|
--task deblurring \
|
||
|
|
--im_path ../../DocScanner/src/rectified/IMG_20260905_172410_rec.png \
|
||
|
|
--out_folder ./output_pipeline
|
||
|
|
|
||
|
|
time python inference.py \
|
||
|
|
--task appearance \
|
||
|
|
--im_path ./output_pipeline/IMG_20260905_172410_rec_deblurring.png \
|
||
|
|
--out_folder ./output_pipeline
|
||
|
|
|
||
|
|
# 3. Appearance
|
||
|
|
time python inference.py \
|
||
|
|
--task appearance \
|
||
|
|
--im_path ./output_pipeline/IMG_20260905_172410_rec_deshadowing.png \
|
||
|
|
--out_folder ./output_pipeline
|
||
|
|
|
||
|
|
time python inference.py \
|
||
|
|
--task dewarping \
|
||
|
|
--im_path /pkg/gnu/unitprint/DocScanner/src/distorted/IMG_20260905_172410.jpg \
|
||
|
|
--out_folder ./output_pipeline
|