经典深度学习推理代码
Ai_Taoism
这个作者很懒,什么都没留下…
展开
-
Detectron2和fvcore中的Registry机制详解
Detectron2和fvcore中的Registry机制详解为什么要有Registry机制如何使用Registry机制Registry机制内部是怎么实现的为什么要有Registry机制Registry机制原本是fvcore框架里面的功能,因为fvcore是一个轻量级的核心库,它提供了在FAIR开发的各种计算机视觉框架(如Detectron2、PySlowFast和ClassyVision)中共享的最常见和最基本的功能。所以Detectron2里面有很多功能都是直接从fvcore框架里面直接拿过来用的。原创 2022-04-24 12:54:51 · 3100 阅读 · 2 评论 -
用pytorch计算神经网络模型需要占用的显存
用pytorch计算神经网络模型需要占用的显存#--------------------------------------------------------------------##作用:计算模型需要占用的显存,方便知道显卡够不够用#使用方法:将模型初始化之后,传入Calculate_gpu_memory()即可#--------------------------------------------------------------------#import torchimport原创 2021-09-28 19:46:23 · 2106 阅读 · 0 评论 -
pytorch实现DETR的推理程序
pytorch实现DETR的推理程序为什么要写这个代码推理结果比较为什么要写这个官方代码在https://github.com/facebooksearch/detr,但是官方代码没有提供推理代码,只提供了训练和评估的代码,所以我在这里写一下推理代码。代码#这个文件是自己写的import argparseimport randomimport timefrom pathlib import Pathimport numpy as npimport torchfrom models im原创 2021-08-08 20:32:12 · 4643 阅读 · 8 评论