プログラミング初心者です。今、以下のコードを教材通りに実行したのですが、Nameerrorが発生して調べたのですが、解決方法が分かりませんでした。ご教示いただけないでしょうか?
python
1 2class Detect(Function): 3 4 def __init__(self,conf_thresh=0.01,top_k=200,nms_thresh=0.45): 5 self.softmax=nn.Softmax(dim=-1) 6 self.conf_thresh=conf_thresh 7 self.top_k=top_k 8 self.nms_thresh=nms_thresh 9 def forward(self,loc_data,conf_data,dbox_list): 10 num_batch=loc_data.size(0) 11 num_dbox=loc_data.size(1) 12 num_classes=conf_data.size(2) 13 conf_data=self.softmax(conf_data) 14 output=torch.zeros(num_batch,num_classes,self,top_k,5) 15 conf_preds=conf_data.transpose(2,1) 16 for i in range(num_batch): 17 decode_boxes=decode(loc_data[i],dbox_list) 18 conf_scores=conf_preds[i],clone() 19 for cl in range(1,num_classes): 20 c_mask=conf_scores[cl].gt(self.conf_thresh) 21 scores=conf_scores[cl][c_mask] 22 if scores.nelement()==0: 23 continue 24 l_mask=c_mask.unsqueeze(1).expand_as(decode_boxes) 25 boxes=decode_boxes[l_mask].voew(-1,4) 26 ids,count=nm_suppression( 27 boxes,scores,self.num_thresh,self.top_k 28 ) 29 output[i,cl,:count]=torch.cat((scores[ids[:count]])) 30
NameError Traceback (most recent call last)
<ipython-input-47-4f21339312cc> in <module>()
----> 1 class Detect(Function):
2
3 def init(self,conf_thresh=0.01,top_k=200,nms_thresh=0.45):
4 self.softmax=nn.Softmax(dim=-1)
5 self.conf_thresh=conf_thresh
NameError: name 'Function' is not defined
回答2件
あなたの回答
tips
プレビュー