pytorch的mask-rcnn的模型参数解释

2024-04-23 07:16:04
开发
37

输入图像1920x1080,batch_size=8为例.

训练阶段

loss_dict = model(images,targets)

入参
- images: List(Tensor(3,1920,1080))[8]
- targets: List(dict()[3])[8] dict详情见下表:

key	type	dtype	size	remark
boxes	Tensor	float32	(n,4)¹	the ground-truth boxes in [x1, y1, x2, y2] format, with 0 <= x1 < x2 <= W and 0 <= y1 < y2 <= H.
labels	Tenosr	int64	(n,)	the class label for each ground-truth box
maskes	Tensor	uint8	(n,1920,1080)[N,H,W]	the segmentation binary masks for each instance,实际就是0和1,有对象的区域就是1,否则就是0,这个照片有多少个对象就有多少个mask
area*	Tensor	float32	(n,)	对象面积
iscrowd*	Tensor	int64	(n,)	是否为一群对象(coco数据集会标注)
image_id*	int			图像编号

*为非必要参数,有一些数据集合处理的时候会标注上去*

返回 loss_dict dict()[5] dict详情见下表:

key	type	dtype	size	损失函数	remark
loss_classifer	Tensor	float32	()	CrossEntropyLoss	对象分类损失（Classification Loss）：
loss_box_reg	Tensor	float32	()	Smooth L1 Loss/MSE	边界框回归损失（Bounding Box Regression Loss）：
loss_mask	Tensor	float32	()	Binary Cross-Entropy Loss	掩膜损失（Mask Loss）：
loss_objectness	Tensor	float32	()	CrossEntropyLoss	RPN分类损失（RPN Classification Loss）：前景/背景二分类损失
loss_rpn_box_reg	Tensor	float32	()	Smooth L1/MSE	RPN边界框回归损失（RPN Bounding Box Regression Loss）

推理阶段

predict = model(images)

入参 images: List(Tensor(3,1920,1080))[8] 不变
返回 predict:List(dict()[4])[8] dict详情见下表:

key	type	dtype	size	remark
boxes	Tensor	float32	(m,4)²	the predicted boxes in [x1, y1, x2, y2] format,预测的所有的边界框
labels	Tensor	int64	(m,)	the predicted labels for each instance
boxes	Tensor	float32	(m,)	the scores or each instance
boxes	Tensor	float32	(m,1,1920,1080)[M, 1, H, W]	the predicted masks for each instance, in 0-1 range. In order to obtain the final segmentation masks, the soft masks can be thresholded, generally with a value of 0.5 (mask >= 0.5).实际存储的是一个软掩膜,0.5以下的也有,存在比较平滑的过度

参考官方文档
maskrcnn_resnet50_fpn

n为此图像上的实例个数 ↩︎
m为此图像 ↩︎

原文地址:https://blog.csdn.net/qq_37293230/article/details/138081000 本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：https://www.suanlizi.com/kf/1782548955314393088.html 如若内容造成侵权/违法违规/事实不符，请联系《酸梨子》网邮箱：1419361763@qq.com进行投诉反馈，一经查实，立即删除！

阅读全部

pytorch的mask-rcnn的模型参数解释

训练阶段

loss_dict = model(images,targets)

推理阶段

predict = model(images)

相关推荐

最近更新

热门阅读