DexiNed 실습(1) - 모델 테스트

이미지의 경계선을 그려주는 모델같은데..... 테스트 해봄

 

1. tensorflow 1.15.4 버전에서 구동할거라

   - cuda : 10.0

   - cudnn : 7.4로 맞춤

 

2. anaconda 에서 환경을 만든다

conda create -n dexined python=3.7
conda activate dexined

 

3. 활성화 시킨 다음 tensorflow설치한다

pip install tensorflow-gpu==1.15.4

 

4. github에서 소스를 받은 후 legacy폴더로 이동

git clone https://github.com/xavysp/DexiNed
cd DexiNed/legacy

 

5. requirements.txt파일을 이용하여 설치 시작해야 하는데, 그냥 설치하면 다음과 같이 오류 발생

** 그러나 그대로 설치하면 다음과 같이 오류 발생

requirements.txt파일을 다음과 같이 수정

....
grpcio==1.26.0
h5py==2.10.0
idna==2.9
Keras-Applications==1.0.8
Keras-Preprocessing==1.1.0
Markdown==3.1.1
matplotlib==3.1.2
mock==4.0.2
numpy==1.18.1
oauthlib==3.1.0
opencv-python==4.4.0   =(수정)=> opencv-python==4.4.0.46
opt-einsum==3.2.1
pkg-resources==0.0.0    <== 삭제
protobuf==3.11.2
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyparsing==2.4.6
python-dateutil==2.8.1
requests==2.23.0
requests-oauthlib==1.3.0
.....

후에 다음과 같이 설치

pip install -r requirements.txt

 

6. 추가로 다음을 설치해준다

pip install pillow

 

7. 다음을 실행하면 오류 발생한다

python run_model.py --image_width=512 --image_height=512

8. 아무이름.pth파일을 만들어서 github를 다운받은 경로를 넣고

'F:\D\DL\DexiNed'

[Anaconda설치경로]\envs\dexined\Lib\site-packages 에 복사한다

** https://doitgrow.com/24 참조

 

9. https://drive.google.com/drive/folders/1fLBpOrSXC2VOWUvDtNGyrHcuB2IB-4_D 경로에서 파일 4개를 다운받는다

1) train_2.zip파일을 풀어서 legacy/checkpoints/DXN_BIPED/train 폴더를 차례로 만들어서 넣어준다

 2) data_list.zip파일을 풀어서 legacy/data폴더를 만들고 그 안에 데이터 파일을 넣는다

10. 다음의 명령어로 테스트를 진행한다

python run_model.py --image_width=512 --image_height=512 --model_state=test

뭔가 진행이 된다.

** legacy\results\edges\DexiNed_BIPED2CLASSIC 폴더에 4개의 폴더가 생성되고, 각각의 폴더에 결과가 저장되어 있다