New virsion for multi-categories were available. Since too many changes above old virsion, please refer to this site https://github.com/ChenWWWeixiang/diagnosis_covid19
I notice that grad_volume.py, line 279, 280 img is reading from the path of mask, and seg is reading from the path of image, which would cause error for the next lines.
Could you check and correct it if I am right :)
Thanks for sharing the code.