随笔分类 - pytorch
摘要:今天整了一个unet网络 通道数一直不对,找不出bug,一晚上了,烦死了!!!
阅读全文
摘要:VOC数据集 官网网站 镜像网站 VOC 2007 VOC 2012 介绍 PASCAL VOC 挑战赛主要有 Object Classification 、Object Detection、Object Segmentation、Human Layout、Action Classification
阅读全文
摘要:pytorch训练CIFAR10数据集
阅读全文
摘要:自定义一个数据集 from torch.utils.data import Dataset import os import cv2 # 定义一个类,继承Dataset class MyData(Dataset): def __init__(self, root_dir, label_dir): s
阅读全文