GithubHelp home page GithubHelp logo

FLOPS about yolo-fastestv2 HOT 2 OPEN

dog-qiuqiu avatar dog-qiuqiu commented on July 21, 2024
FLOPS

from yolo-fastestv2.

Comments (2)

Lebhoryi avatar Lebhoryi commented on July 21, 2024

谷歌一下,网上都能查到,一个是自己计算,一个是所用的神经网络框架自己计算得到

from yolo-fastestv2.

noahzhy avatar noahzhy commented on July 21, 2024

you can get the flops of you model as follows:

    from ptflops import get_model_complexity_info

    h, w = (192, 192)

    with torch.cuda.device(0):
        macs, params = get_model_complexity_info(
            model, (3, h, w),
            as_strings=True, print_per_layer_stat=True, verbose=True
        )
        flops = str(float(macs.split(' ')[0]) * 2) + " GFLOPs"
        print('{:<30}  {:<8}'.format('Computational complexity: ', macs))
        print('{:<30}  {:<8}'.format('FLOPs: ', flops))
        print('{:<30}  {:<8}'.format('Number of parameters: ', params))

from yolo-fastestv2.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.