Attention-based-Dual-Branch-Network-

A Multi-label Sewer Defects Classification Methodology Using Attention-based Dual Branch network

Training with label mask

python main.py --use_lmt

python main.py

python inference.py

python calculate_results.py

Name		Name	Last commit message	Last commit date
Latest commit History 28 Commits
Annotations		Annotations
adjacency_matrices		adjacency_matrices
dataloaders		dataloaders
models		models
utils		utils
word_embeddings		word_embeddings
README.md		README.md
TrainEmbeddings.ipynb		TrainEmbeddings.ipynb
TrainEmbeddings.txt		TrainEmbeddings.txt
calculate_results.py		calculate_results.py
config_args.py		config_args.py
ideas.ipynb		ideas.ipynb
inference.py		inference.py
load_data.py		load_data.py
main.py		main.py
metrics.py		metrics.py
optim_schedule.py		optim_schedule.py
run_epoch.py		run_epoch.py