search

FengheTan9 / Multi-Level-Global-Context-Cross-Consistency

Official Pytorch Code base for "Multi-Level Global Context Cross Consistency Model for Semi-Supervised Ultrasound Image Segmentation with Diffusion Model"
MIT License
33 stars 1 forks source link

inference code #3

Open godfunc123 opened 1 year ago

godfunc123 commented 1 year ago

Is there a code for the inference ？especially for the experiment analyzed the average inference time and GFLOPs, Thank you

FengheTan9 commented 1 year ago

Thanks for using our code. You can use the following code for analyzing: from fvcore.nn import FlopCountAnalysis, parameter_count_table flops = FlopCountAnalysis(model, input) or from thop import profile flops, params = profile(model, inputs=(input, ))

godfunc123 commented 1 year ago

Thank you