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 · 1744 阅读 · 1 评论 -
pytorch手写VGG16网络,两种写法,低阶基础写法
1、vgg16的网络结构图2、vgg16网络分析输入的数据大小为[224*224]的图片,通道数为3,共有13个卷积层,3个全联接层,所以叫vgg16。具体的操作完之后的大小见代码注释。3、代码注释import torchvisionfrom torch import nnfrom torch.nn import MaxPool2dclass VGG16(nn.Module): def __init__(self): super(VGG16, self).__in原创 2021-09-16 09:32:43 · 3486 阅读 · 1 评论