TunnelDiff — Official Pytorch Implementation

Diffusion Model for Tunnel Lining Image Enhancement

This repository contains the official TensorFlow implementation of the following paper:

TunnelDiff: A Brightness-Guided Image Restoration Diffusion Model for Enhancing Defect Detection in Low-Quality Tunnel Lining Images

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
assets		assets
callback		callback
config		config
dataset		dataset
logger		logger
metric		metric
model		model
util		util
.gitignore		.gitignore
README.md		README.md
test_exposure_errors.py		test_exposure_errors.py
test_exposure_errors.sh		test_exposure_errors.sh
test_out_exposure_errors.py		test_out_exposure_errors.py
test_out_tunnel_defect.py		test_out_tunnel_defect.py
test_tunnel_defect.py		test_tunnel_defect.py
test_tunnel_defect.sh		test_tunnel_defect.sh
test_tunnel_defect_enhance.py		test_tunnel_defect_enhance.py
test_tunnel_defect_enhance.sh		test_tunnel_defect_enhance.sh
test_tunnel_defect_low0.py		test_tunnel_defect_low0.py
train.py		train.py
train.sh		train.sh

Navigation Menu