
pytorch
大鹏要高飞
深度学习,机器视觉,目标检测
展开
-
错误detectron2.data.detection_utils.SizeMismatchError: Mismatched image shape for image
detectron2.data.detection_utils.SizeMismatchError: Mismatched image shape for image data/augmented_images/PAID/0062_2.jpg, got (864, 1152), expect (1152, 864). Please check the width/height in your annotation. 解决办法: # 你的报错信息会提示你detectron2/data/detection_u原创 2021-11-30 22:10:06 · 1935 阅读 · 1 评论 -
pytorch手写VGG16网络,两种写法,低阶基础写法
1、vgg16的网络结构图 2、vgg16网络分析 输入的数据大小为[224*224]的图片,通道数为3,共有13个卷积层,3个全联接层,所以叫vgg16。具体的操作完之后的大小见代码注释。 3、代码注释 import torchvision from torch import nn from torch.nn import MaxPool2d class VGG16(nn.Module): def __init__(self): super(VGG16, self).__in原创 2021-09-16 09:32:43 · 3666 阅读 · 1 评论