search

ristea / sspcab

Other
147 stars 15 forks source link

A question about HF2VAD #6

Open tylgczq opened 2 years ago

tylgczq commented 2 years ago

Thank you for your great work, I have a question:

How to add sspcab to HF2VAD，can you send the modified code？

tylgczq commented 2 years ago

class up(nn.Module): def init(self, in_ch, out_ch, bilinear=False, op="none"): super(up, self).init() self.bilinear = bilinear self.op = op if self.bilinear: self.up = nn.Sequential(nn.Upsample(scale_factor=2, mode='bilinear', align_corners=True), nn.Conv2d(in_ch, in_ch // 2, 1), ) else: self.up = nn.ConvTranspose2d(in_channels=in_ch, out_channels=in_ch // 2, kernel_size=3, stride=2, padding=1, output_padding=1) assert op in ["concat", "none"]

    if op == "concat":
        self.conv = double_conv(in_ch, out_ch)
    else:
        self.conv = double_conv(out_ch, out_ch)

whether only need to modify the last piece of code？ self.conv = double_conv(out_ch, out_ch) Add sspcab here？