chundoong-lab-ta/APWS23/project/README.md

35 lines
812 B
Markdown
Raw Normal View History

2023-02-01 22:30:00 +09:00
# 2023-winter-school-project
2023년 2월 겨울학교 프로젝트
2023-02-06 17:10:16 +09:00
참고 git : https://github.com/milesial/Pytorch-UNet
# 실행
2023-02-06 17:22:03 +09:00
## 모델 가중치 준비
2023-02-06 17:21:14 +09:00
스크립트를 통해 weights.bin 다운로드
```
sh download_weights.sh
```
2023-02-06 17:22:03 +09:00
## 이미지를 바이너리로 변환 (img2bin)
2023-02-06 17:10:16 +09:00
jpg이미지를 바이너리 파일로 생성
```
2023-02-06 17:26:50 +09:00
python tools/img2bin.py img/inData1.jpg input1.bin
2023-02-06 17:10:16 +09:00
```
2023-02-06 17:22:03 +09:00
## C코드 컴파일 및 실행
2023-02-06 17:10:16 +09:00
```
make clean && make -j
./main weights.bin input1.bin output1.bin
```
2023-02-06 17:22:03 +09:00
## 바이너리 결과를 이미지로 변환 (bin2img)
2023-02-06 17:10:16 +09:00
바이너리로 저장된 추론 결과를 png이미지로 변환
```
2023-02-06 17:26:50 +09:00
python tools/bin2img.py output1.bin output1.png
2023-02-06 17:10:16 +09:00
```
2023-02-06 17:23:23 +09:00
# ETC
## Python 환경 준비
python실행에 필요한 numpy, pillow, pytorch env 설치
```
2023-02-06 17:26:50 +09:00
conda env create -f ./tools/env.yaml
2023-02-06 17:23:23 +09:00
```