cannonvirus/Unet_baseline

프로젝트명

Anormal 이미지 비교로 승가 및 기타 이상행위 잡아내는 프로젝트

설치 방법

Docker:

intflow/de-blur-competition:HINet

Utils

utils/unet_datamaker.py

normal 데이터 셋과 anormal 데이터 셋을 합치는 역할

실행코드

source /opt/conda/bin/activate
/opt/conda/bin/python unet_datamaker.py

config.yaml 설명

📙 : path variable

• normal_path : normal timeseries dataset path
• anormal_path : anormal timeseries dataset path
• out_path : output path

📗 : option

• normal_sampling : normal_path에서 몇개를 샘플링 할 것인가?
• anormal_sampling : anormal_path에서 몇개를 샘플링 할 것인가? 만약 -1이라면, 모두 선택
• only_normal : Enc-Dec trainset 만들 때 사용 (정상데이터로만 만들고 싶을 때)

---

utils/augmentor.py

주로 anormal data 증강용

실행코드

source /opt/conda/bin/activate
/opt/conda/bin/python augmentor.py

config.yaml 설명

📙 : path variable

• unet_data_path : timeseries dataset path
• output_path : output path

📗 : option

• max_copy_img : 한 source에 대해서 몇번 gen 할것인가?
• original_copy : copy origin_repo to output_path
• max_algorithm : select Algorithm number
• option_algorithm : 비정형 Algorithm 포함할 것인가? ~ (비 뿌리기, 눈 뿌리기 등)

Predict

predict/test_one.py

Sample 하나 넣었을 때 PSNR과 예측이미지, Anormal prob 관측

실행코드

source /opt/conda/bin/activate
/opt/conda/bin/python test_one.py

config.yaml 설명

📙 : path variable

• input_path : (String) One folder timeseries dataset path
• model_path : (String) Model path

📗 : option

• scale : (Default, INT) 96 | (half_model, INT) 48
• threshold : (float) anormal threshold [Not Use]
• bgr2rgb : (bool) make img RGB style

---

predict/test_PSNR.py

Test dataset을 넣었을 때 anormal layer에 대한 Confusion matrix

실행코드

source /opt/conda/bin/activate
/opt/conda/bin/python test_PSNR.py

config.yaml 설명

📙 : path variable

• input_path : (String) Many folder timeseries dataset path
• model_path : (String) Model path

📗 : option

• scale : (Default, INT) 96 | (half_model, INT) 48
• threshold : (float) anormal threshold [Not Use]
• bgr2rgb : (bool) make img RGB style

개발 환경 설정

3090 X 8GPU

업데이트 내역

• None

정보

주소현 – cannonvirus@intflow.ai