UnsupervisedDeepImageStitching 논문 소스 돌려보기

소스구동 환경

os : ubuntu 16.04

python 3.6

tensorflow 1.13.1

 

docker 로 위의 소스를 돌려 보려고 할 때 기존에 docker 에서 제공하는 공식 이미지 패키지는 Tensorflow 1.13.1 과 python 3.5 가 한 세트로 묶여 있어서 주의해야 함.

 

1. 논문 링크 :

https://arxiv.org/pdf/2106.12859v1.pdf

2. 소스 github link : 

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

GitHub - nie-lang/UnsupervisedDeepImageStitching: TIP2021 - Unsupervised deep image stitching network

3. 소스 수정

1. ImageAlignment/Codes/inference.py 파일 수정 / 실행

$ cd DeepImageStitching-1.0-main/ImageAlignment/Codes/ 
$ python inference.py

 

위의 inference.py 파일에서 length 는 데이터셋의 testing 폴더(우리가 stitching 을 진행할 파일들이 모여있는 곳) 의 사진의 개수이다.

2. ImageReconstruction/Codes/inference.py 파일 수정 / 실행

$ cd /DeepImageStitching-1.0-main/ImageReconstruction/Codes/ 
$ python inference.py

위의 inference.py 파일에서 length 는 데이터셋의 testing 폴더(우리가 stitching 을 진행할 파일들이 모여있는 곳) 의 사진의 개수이다.

2. ImageReconstruction/Codes/output_inference.py 파일 수정 / 실행

$ cd /DeepImageStitching-1.0-main/ImageReconstruction/Codes/ 
$ python output_inference.py

length 의 기본값이 10440 으로 되어 있는데 이를 1106(testing 폴더의 파일 개수) 로 고쳐준다

DataLoader(train_foler) 를 DataLoader(test_folder) 로 변경한다