计算机视觉相关库整理

前言

前前后后配置过多次环境，每次装库都会遇到各种各样的问题，不同库的安装命令虽然总体并无太大差别，但是库名繁杂，因此决定整理一番方便日后安装需要。

镜像

因为pip下载服务器在国外，经常因为网速问题下载失败，这里可以使用国内镜像，例如清华大学镜像

pip install –i https://pypi.tuna.tsinghua.edu.cn/simple _packagename_

模型统计类

ptflops

pip install ptflops

使用例子

import torchvision.models as models
import torch
from ptflops import get_model_complexity_info

with torch.cuda.device(0):
  net = models.densenet161()
  macs, params = get_model_complexity_info(net, (3, 224, 224), as_strings=True,
                                           print_per_layer_stat=True, verbose=True)
  print('{:<30}  {:<8}'.format('Computational complexity: ', macs))
  print('{:<30}  {:<8}'.format('Number of parameters: ', params))