search

LeeJunHyun / Image_Segmentation

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.
2.66k stars 594 forks source link

> hey guys. use pytorch<=1.2.0 (not confirmed) or change the funtion in "evaluation" to fit the calculation of **bool tensor** will sovle the problem #83

Closed rouchoo closed 1 year ago

rouchoo commented 2 years ago

hey guys. use pytorch<=1.2.0 (not confirmed) or change the funtion in "evaluation" to fit the calculation of bool tensor will sovle the problem e.g. change "TP = ((SR==1)+(GT==1))==2" to "TP = SR & GT"

thanks, i try it now, and do you have some recommend with pytorch > 1.2?

_Originally posted by @CodeXiaoLingYun in https://github.com/LeeJunHyun/Image_Segmentation/issues/58#issuecomment-741368638_

rouchoo commented 2 years ago

when TP = ((SR==0)+(GT==1))==2 and TP = ((SR==1)+(GT==0))==2，how can i change？