search

YuzhouPeng / unet-tensorrt

this is a tensorrt version unet, inspired by tensorrtx
MIT License
36 stars 11 forks source link

the output of the cuda context,the using simoid? #11

Open henbucuoshanghai opened 3 years ago

henbucuoshanghai commented 3 years ago

https://github.com/YuzhouPeng/unet-tensorrt/blob/a7a6d81ddf1ed553a87d1bdbd83075c6b892dacb/unet.cpp#L363

why using the sigmoid?

YuzhouPeng commented 2 months ago

https://github.com/wang-xinyu/tensorrtx/tree/master/unet please use newest repo for testing, this old repo is no longer maintained