DeepRectangling Image Stitching 논문 소스 돌려보기

소스구동 환경

os : ubuntu 16.04

python 3.6

tensorflow 1.13.1

 

1. 논문 링크 : 

https://arxiv.org/pdf/2203.03831.pdf

2. 소스 github link : 

https://github.com/nie-lang/DeepRectangling

GitHub - nie-lang/DeepRectangling: CVPR2022 (Oral) - Deep Rectangling for Image Stitching: A Learning Baseline

3. 소스 수정

/Codes 에 DIR-D 데이터 셋 폴더를 올린다.

* /Codes/constant.py 에서 TRAIN_FOLER 와 TEST_FOLDER 를 DIR-D 데이터셋에 맞게 경로를 수정한다.

* GPU 사용 index 를 자신에 맞게 수정한다.

 

4. 소스 실행

4.1. Training

cd Codes/
python train.py

4.2. Testing

1. 기존 test data set 으로 test 진행시

cd Codes/
python inference.py

2. 자신의 데이터 셋으로 test 를 진행하고 싶을 때

cd Codes_for_Arbitrary_Resolution/
python inference.py

여기서는 Codes_for_Arbitrary_Resolution 폴더 안에서 위에 설명한 것처럼 constant.py 를 수정해준다.

여기서 other_dataset 폴더 안에 자신이 만들어 둔 mask, input jpg 파일을 각각 mask, input 폴더 안에 넣어준다.

그 후 python inference.py 를 실행한다.