$ python main.py --helpusage: main.py [-h] [--method-names METHOD_NAMES [METHOD_NAMES ...]]               [--mode {ops_params,fps,gpu_mem} [{ops_params,fps,gpu_mem} ...]]A simple toolkitfor counting the FLOPs/MACs, Parameters, FPS and GPU Memory of the model.optional arguments:-h, --help            show thishelp message andexit  --method-names METHOD_NAMES [METHOD_NAMES ...]                        The names of the methods you want to evaluate.  --mode {ops_params,fps,gpu_mem} [{ops_params,fps,gpu_mem} ...]

Usage

pip install fvcore

python main.py --method-names zoomnet ugtr c2fnet ujsc pfnet mgl_r slsr sinet

Change Log

@misc{MethodsCmp,author       = {Youwei Pang},title        = {MethodsCmp: A Simple Toolkit for Counting the FLOPs/MACs, Parameters and FPS of Pytorch-based Methods},howpublished = {\url{https://github.com/lartpang/MethodsCmp}},year         = {2021}}