- 博客(5)
- 收藏
- 关注
RSS订阅原创 pytorch 定义dice_coeff, SoftDiceLoss,BCELoss2d损失函数
import numpy as npimport torch.nn as nnimport torch.nn.functional as Fdef dice_coeff(pred, target): smooth = 1. m1 = pred.view(num, -1) m2 = target.view(num, -1) intersection = (m1*m2).sum() return (2. * intersection + smooth) /.
2021-08-05 15:07:22
2558
原创 目标检测中，计算IoU值
1. # 计算IoU，矩形的坐标形式为xyxydef box_iou_xyxy(box1, box2): # 获取box1左上角和右下角的坐标 x1min, y1min, x1max, y1max = box1[0], box1[1], box1[2], box1[3] # 计算box1的面积 s1 = (y1max - y1min + 1.) * (x1max - x1min + 1.) # 获取box2左上角和右下角的坐标 x2min, y2min,
2021-07-04 22:56:26
568
1
原创 opencv-python 读取和保存图像
1. 当路径全为英文时，采用的图像读取和保存方式import cv2# 读取图像img = cv2.imread("./raw.jpg", 0) # 0: gray image;1: bgr 图像; -1: 图像原格式输出# 保存图像cv2.imwrite("./save.jpg", img)2. 当路径不全为英文时，即路径中可能存在中文，采用的图像读取和保存方式# 读文件， 这种方式能读取路径中含有中文的图像文件def imread(path): # image_arra
2021-07-04 22:42:02
1332
原创 opencv-python 读取和保存视频
# 1. capture video from camera# 通过相机捕获视频流。opencv中通过 VideoCapture 对象捕获视频，其参数是相机设备索引号或是视频文件的名字。# 本地设备为0，如果你还有其他的相机设备，可设为1,2，...# 可通过cap.isOpened() 检查相机是否初始化，有可能有些错误是我能成功初始化造成的import numpy as npimport cv2cap = cv2.VideoCapture(0)# frame_width = ca
2021-07-04 22:02:35
728
原创 pytorch 中，使用自定义的损失函数训练模型
pytorch 中，使用自定义的损失函数训练模型# 当Tensor算数操作（+， -，*， %等）中，有一个Tensor的resquire_grad=True,则该操作得到的Tensor# 能够实现反向传播，自动求导的功能。# 所以第一种自定义loss函数的方法就是使用tensor的math operation实现loss定义。###############################################################1. 继承于nn.Module，nn.Mod
2021-07-04 20:34:27
1256
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人