def forward(self, x):
    spatial_out = self.spatial_path(x)
    (context16, context32) = self.context_path(x)
    fm_fuse = self.ffm(spatial_out, context16)
    output = self.output_head(fm_fuse)
    if self.training:
        context_out16 = self.context16_head(context16)
        context_out32 = self.context32_head(context32)
        return (output, context_out16, context_out32)
    return output
def forward(self, x):
    spatial_out = self.spatial_path(x)
    (context16, context32) = self.context_path(x)
    fm_fuse = self.ffm(spatial_out, context16)
    output = self.output_head(fm_fuse)
    if self.training:
        context_out16 = self.context16_head(context16)
        context_out32 = self.context32_head(context32)
        return (output, context_out16, context_out32)
    return output