image-segmentation-keras icon indicating copy to clipboard operation
image-segmentation-keras copied to clipboard

UNet 3+ model added

Open hamidriasat opened this issue 2 years ago • 2 comments

Two changes

  • New model UNet 3+ added
  • Reference also added in README.md

hamidriasat avatar Oct 17 '22 20:10 hamidriasat

Hi thanks for the PR, but it seems like that the new model you added might not work with other functionality of the repo. For e.g. it is not added in the all_models.py , hence cli tools wont be able to use it.

divamgupta avatar Nov 05 '22 00:11 divamgupta

Hi @divamgupta, thanks for your reply. New changes have been made in code to make it compatible with rest of the repo functionality. Please test it with cli tools. I hope it works this time, if anything is missing please let me know.

Following changes have been made in the current commit:

  • Unet3+ model reference added in all_models.py
  • Code formatting in unet3_plus.py to make sure local methods are not accessible to outside classes/objects
  • PEP8 code formatting of test_models.py to make sure it's easily readable
  • Unet3+ reference added in test_models method of test_models.py file

hamidriasat avatar Nov 06 '22 13:11 hamidriasat