Piv 58b8e53986 Fix packnet residual block and layers, refactor to support different amount of residual layers ...

I noticed the number of parameters didn't match up to the paper (~128 million)
Fixed this by doing the following:
 - Kernel size of 1 for 3rd conv2d in residual block
 - Use add rather than concat in residual block
 - Fixed add/concat features in decode layers
 - Fixed final layers -> this also allows features_3d == 16 to work

2021-08-07 21:00:17 +09:30

5.7 KiB

Raw Blame History

View Raw