| 71 | }; TORCH_MODULE(ASPP); |
| 72 | |
| 73 | class DeepLabV3DecoderImpl : public torch::nn::Module |
| 74 | { |
| 75 | public: |
| 76 | DeepLabV3DecoderImpl(int in_channels, int out_channels = 256, std::vector<int> atrous_rates = { 12, 24, 36 }); |
| 77 | torch::Tensor forward(std::vector< torch::Tensor> x); |
| 78 | int out_channels = 0; |
| 79 | private: |
| 80 | torch::nn::Sequential seq{}; |
| 81 | }; TORCH_MODULE(DeepLabV3Decoder); |
| 82 | |
| 83 | class DeepLabV3PlusDecoderImpl :public torch::nn::Module { |
| 84 | public: |
nothing calls this directly
no outgoing calls
no test coverage detected