crack_segmentation

crack_segmentation copied to clipboard

the pretrained weight of resnet is invalid

6

I want to cite your work on github in my paper related to UAV bridge inspection, but the weight of the pretrained weight of resnet is invalid, please check it...

KangchengLiu

the link of pretraind weight of Resnet is invaild, pls resent agina.

2

YurongChen1998

in the utils.py

1

in line number 30 use this code to fix it def cuda(x): return x.cuda(non_blocking=True) if torch.cuda.is_available() else x

Chirag0096

Bro in the code their is problem

1

use this for inference :- python inference_unet.py -img_dir ./test_images -model_path ./model.pth -model_type resnet101 -out_viz_dir ./viz_result -out_pred_dir ./pred_result -threshold 0.5

Chirag0096

Where does the dataset Rissbilder_for_Florian come from？

1

Rissbilder_for_Florian dataset is in your data，but I can't find its source. Can you show it? Thank you very much.

Gu-EH

Garhoode