- (2022-07 ~ 2023-02) 광운대학교 산학연계 SW프로젝트 : 사미용두팀
자동차 부품 제조 현장의 학습/검증용 데이터 불균형을 해결하기 위한 GAN 기반의 불량 데이터 생성
SMYD
|- data
| |- L_01-1 ~ L_01-6 # Any images
|- models
| |- generator.py
| |- discriminator.py
| |- train.py
|- utils
| |- preprocess.py
| |- dataloader.py
|- results
| |- scoring.py
|- main.py
|- requirements.txt
| ...
Python 3.9.5
OpenCV 4.6.0.66
Pytorch 1.12.1
Torchvision 0.13.1
Numpy 1.22.3
Scipy 1.7.1
- node.js
- npm
- nodemon
$ git clone
$ npm install
$ npm run start
사전 데이터
↓
(1) 리크 추출
(2) 그라드라인 제거
(3) 리사이즈
(4) 리크 데이터 증강
↓
학습 데이터
출처 : DCGAN
정상 데이터
↓
(1) 모델 불러오기
(2) 리크 생성
(3) 사용자 설정 확인
(4) 정상 데이터에 합성
↓
불량 데이터
이미지 파일 업로드
생성 위치 랜덤 지정 생성 위치 사용자 지정 여러개 리크 생성
불량 이미지 다운로드
메인 화면 | 사용자 설정 화면 | 불량 이미지 생성 화면 |
---|---|---|
JavaScript | CSS | HTML | Node.JS | Express | Python | Pytorch |
---|---|---|---|---|---|---|