Skip to content

Y
yolov7-obb

白杨攻城狮 / yolov7-obb
与 Fork 源项目一致

从无法访问的项目Fork

通知 1
Star 0
Fork 0
Y
yolov7-obb

前往新版Gitcode，体验更适合开发者的 AI 搜索 >>

提交 9a1f858e 编写于 作者: Bubbliiiing's avatar Bubbliiiing
浏览文件
操作

fix __int64 bug

上级 01e1c8fa
隐藏空白更改
内联 并排
Showing with 2 addition and 1 deletion
nets/yolo_training.py
浏览文件 @ 9a1f858e
...@@ -448,6 +448,7 @@ class YOLOLoss(nn.Module): ...@@ -448,6 +448,7 @@ class YOLOLoss(nn.Module):
# anchors_i [num_anchor, 2] # anchors_i [num_anchor, 2]
#----------------------------------------------------# #----------------------------------------------------#
anchors_i = torch.from_numpy(self.anchors[i] / self.stride[i]).type_as(predictions[i]) anchors_i = torch.from_numpy(self.anchors[i] / self.stride[i]).type_as(predictions[i])
anchors_i, shape = torch.from_numpy(self.anchors[i] / self.stride[i]).type_as(predictions[i]), predictions[i].shape
#-------------------------------------------# #-------------------------------------------#
# 计算获得对应特征层的高宽 # 计算获得对应特征层的高宽
#-------------------------------------------# #-------------------------------------------#
...@@ -509,7 +510,7 @@ class YOLOLoss(nn.Module): ...@@ -509,7 +510,7 @@ class YOLOLoss(nn.Module):
# a代表属于该特征点的第几个先验框 # a代表属于该特征点的第几个先验框
#-------------------------------------------# #-------------------------------------------#
a = t[:, 6].long() # anchor indices a = t[:, 6].long() # anchor indices
indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1))) # image, anchor, grid indices indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid indices
anchors.append(anchors_i[a]) # anchors anchors.append(anchors_i[a]) # anchors
return indices, anchors return indices, anchors
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或