weixin_42184622的博客

创建数据读取类自己创建的类需要继承 data.Dataset，同时必须要重载__getitem__以及 __len__函数import torch.utils.data as dataclass PNetDataSet(data.Dataset): def __init__(self, imgs, transforms=None): ''' 目标：获取所有...

1.opencv保存视频文件很小，没有视频cap = cv2.VideoCapture(args.video_path)cv2.namedWindow("SSD", 0)print(cap.get(cv2.CAP_PROP_FRAME_WIDTH),cap.get(cv2.CAP_PROP_FRAME_HEIGHT))#**这里的size必须与最后保存的图片的size一致，否则保存的文件很...

Smooth L1 Loss（Huber）公式调用方法import torch.nn.functional as Fa=torch.tensor([[0,0.5,2,3],[0,1,2,3],[0,1,2,3]],dtype=torch.float32)b=torch.tensor([[1,0.3,3,4],[0,1.2,2.2,3],[0,1,2,5]],dtype=torch.fl...

#111

Pytorch 保存加载模型时的坑Pytorch 保存加载模型时的坑方法1：保存模型的参数和结构信息官方推荐的方法，只保存和恢复模型中的参数填坑如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能，丰富你的文章UML 图表FLowch...