Chirag0096

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

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