Git Product home page Git Product logo

toolcv's Introduction

分类 模型(backbone)

目标检测+分割

pytorch轻便库(类似于 tensorflow的keras)

pytorch 训练加速库

其他

数据增强库

pycocotools

# 计算 mAP
pip install pycocotools

toolcv

tool for CV

requirements

!pip install mmcv-full==1.3.11 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html 
!pip install openmim  > /dev/null
!mim install mmdet
!pip install fvcore>=0.1
!pip install pytorch-lightning>=1.3.8
!pip install timm>=0.4.12
!pip install pycocotools
!pip install pycuda
!pip install thop # 计算网络的flops

# apex 安装
git clone https://github.com/NVIDIA/apex.git
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
# pip install -v --no-cache-dir --global-option="--pyprof" --global-option="--cpp_ext" --global-option="--cuda_ext" ./
# python setup.py install --cpp_ext --cuda_ext
!pip install https://codechina.csdn.net/wc781708249/toolcv/archive/master.zip
# or
!pip install 'git+https://codechina.csdn.net/wc781708249/toolcv.git'
!pip install 'git+https://codechina.csdn.net/wc781708249/toolcv.git@master' # @切换分支
# or
!python setup.py install
!pip install 'git+https://codechina.csdn.net/wc781708249/toolcv.git' > /dev/null
!pip install mmcv-full==1.3.11 -f https://download.openmmlab.com/mmcv/dist/cu110/torch1.7.0/index.html  > /dev/null
!pip install openmim  > /dev/null
!mim install mmdet  > /dev/null

wanb(类似tensorboard)

# pip install wandb

# Flexible integration for any Python script
import wandb

# 1. Start a W&B run
wandb.init(project='gpt3')

# 2. Save model inputs and hyperparameters
config = wandb.config
config.learning_rate = 0.01

# Model training here# 3. Log metrics over time to visualize performance
wandb.log({"loss": loss})

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.