Yeah sure. I will get back to you after reviewing your code. Give me sometime. But I will suggest you to implement it on radiology images, you can get some from kaggle. It will give more clear picture.
Well this is not radiology images, but you can try on it.