unet3 动态分辨率支持

网络结构代码：

  

   

    

     

    
    

     

      # -*- coding: utf-8 -*-
     
    
   

    

     

    
    

     

      import os
     
    
   

    

     

    
    

     

      import time
     
    
   

    

     

    
    

     

      import numpy as np
     
    
   

    

     

    
    

     

      import cv2
     
    
   

    

     

    
    

     

      import torch
     
    
   

    

     

    
    

     

      import torch.nn as nn
     
    
   

    

     

    
    

     

      import torch.nn.functional as torch_f
     
    
   

    

     

    
    

     

      from torch.nn import init
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def weights_init_normal(m):
     
    
   

    

     

    
    

     

          classname = m.__class__.__name__
     
    
   

    

     

    
    

     
    #print(classname)
     
    
   

    

     

    
    

     
    if classname.find('Conv') != -1:
     
    
   

    

     

    
    

     

              init.normal_(m.weight.data, 0.0, 0.02)
     
    
   

    

     

    
    

     
    elif classname.find('Linear') != -1:
     
    
   

    

     

    
    

     

              init.normal_(m.weight.data, 0.0, 0.02)
     
    
   

    

     

    
    

     
    elif classname.find('BatchNorm') != -1:
     
    
   

    

     

    
    

     

              init.normal_(m.weight.data, 1.0, 0.02)
     
    
   

    

     

    
    

     

              init.constant_(m.bias.data, 0.0)
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      def weights_init_xavier(m):
     
    
   

    

     

    
    

     

          classname = m.__class__.__name__
     
    
   

    

     

    
    

     
    #print(classname)
     
    
   

    

     

    
    

     
    if classname.find('Conv') != -1:
     
    
   

    

     

    
    

     

              init.xavier_normal_(m.weight.data, gain=1)
     
    
   

    

     

    
    

     
    elif classname.find('Linear') != -1:
     
    
   

    

     

    
    

     

              init.xavier_normal_(m.weight.data, gain=1)
     
    
   

    

     

    
    

     
    elif classname.find('Ba
     
    
  
 

文章来源: blog.csdn.net，作者：AI视觉网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/122375415