search

jakeret / tf_unet

Generic U-Net Tensorflow implementation for image segmentation
GNU General Public License v3.0
1.9k stars 748 forks source link

Performing a check to make sure that the segmentation map from the ch… #265

Closed siavashk closed 5 years ago

siavashk commented 5 years ago

…ild class is actually boolean array before performing bitwise logical not

Re #264

jakeret commented 5 years ago

thanks a lot for contributing to the code base (and sorry for leaving the PR open for so long)

siavashk commented 5 years ago

It's OK. Since the last time that we communicated, I have implemented the following quality-of-life enhancements to your code base:

  1. Support for SAME padding so that the input image and output segmentation match in width and height.
  2. Script for loading the trained model directly from the checkpoint as opposed to creating a U-Net model first and then loading weights from the model directory.

I can make these changes in my fork and do a pull request if you have bandwidth to review them.

jakeret commented 5 years ago

@siavashk I just realized that I never replied you - I'm very sorry for that.

I would be very interested in those two extensions - hope your still willing to contribute to this project

siavashk commented 5 years ago

Unfortunately, I left my position and I no longer have access to my fork. But thanks for putting this code out there!