vanvalenlab / deepcell-tf / 1465

86%

push

travis-ci-com

Update panopticnet and fpn for future model export support (#352)

* Replaced TensorProduct layers with Conv layers. Added names to all layers

* Added support for lite feature pyramid networks, which use depthwise convs instead of convs during FPN construction

* Removed extraneous function from panopticnet.py

* Switched the upsampling layers to bilinear interpolation as opposed to nearest neighbor - required for models to be quantized

* Restore fpn semantic_upsample to the original function

* Check ndim before creating semantic head

* Added ability to change the interpolation mode of upsampling2d layers

Co-authored-by: David Van Valen <vanvalen@gmail.com>
Co-authored-by: willgraf <7930703+willgraf@users.noreply.github.com>

52 of 52 new or added lines in 2 files covered. (100.0%)

4952 of 6575 relevant lines covered (75.32%)

4.5 hits per line