Update model/CMFNet.py

model/CMFNet.py

@@ -1,7 +1,5 @@
 import torch
 import torch.nn as nn
-from utils import network_parameters
-from thop import profile
 from model.block import SAB, CAB, PAB, conv, SAM, conv3x3, conv_down
 
 ##########################################################################
@@ -193,18 +191,3 @@ class CMFNet(nn.Module):
         return [x_final + mixed_img, mixed_img, mix_r[1], x1_img, x2_img, x3_img, x_final]
 
 
-if __name__ == "__main__":
-    import time
-    model = CMFNet()
-
-    for idx, m in enumerate(model.modules()):
-        print(idx, "-", m)
-    s = time.time()
-
-    rgb = torch.ones(1, 3, 256, 256, dtype=torch.float, requires_grad=False)
-    out = model(rgb)
-    flops, params = profile(model, inputs=(rgb,))
-    print('parameters:', params)
-    print('flops', flops)
-    print('time: {:.4f}ms'.format((time.time()-s)*10))
-