Detr部署

使用代码及环境配置

代码

使用D:\学习文件\研究生文件\部署相关文件\DETR\代码\test1.ipynb文件

环境配置

1.创建环境

conda create -n detr python==3.x(3.8版本以上)

#安装 PyTorch 1.5+ 和 torchvision 0.6+

conda install -c pytorch pytorch torchvision

2.git文件

git clone https://github.com/facebookresearch/detr.git

3.requirementstxt文件

pip install requirements.txt -i Simple Index

代码流程

1.导包

from PIL import Image

import requests

import matplotlib.pyplot as plt

# %config InlineBackend.figure_format = 'retina'

import torch

from torch import nn

from torchvision.models import resnet50

import torchvision.transforms as T

torch.set_grad_enabled(False);

2.设置类DETRdemo

class DETRdemo(nn.Module):
    def __init__(self, num_classes, hidden_dim=256, nheads=8,