美兰区邓紫棋-CSDN博客

原创 STTran 源码解读(4)：evaluation_recall.py

pred_bbox, pred_class, pred_rel, obj_scores, rel_scores(rel_scores来自模型预测，其余变量均来自gt)目标检测可视化特征通过预训练模型进行推断，关系预测结果来自tranformer编解码器模型。3.当三元组和bbox_iou大于阈值的情况下，返回给定预测的GT匹配列表。——实体特征，联合框特征——实体及联合框表示+语义嵌入表示——2.计算头尾实体bbox_iou是否大于阈值。pred_to_gt:从谓词中匹配GT。获得真实三元组和预测三元组。

2023-09-09 20:56:19 58

空空如也

TA创建的收藏夹 TA关注的收藏夹

TA关注的人

qq_34108497的博客

原创 STTran 源码解读(4)：evaluation_recall.py

原创 faster-rcnn训练自己的数据集出现loss为nan的情况

原创 STTran 源码解读(3)：STTran类

原创 STTran 源码解读(2)：detector类

原创 STTran 源码解读(1)：AG类

原创 pytorch安装记录

空空如也

空空如也