search

Toma0916 / GlobalWheatDetection

3 stars 0 forks source link

debug with sanity bboxes #162

Closed kminoda closed 3 years ago

kminoda commented 3 years ago

例のbboxのバグ直した

kminoda commented 3 years ago

bboxない時は、model.pyのl.175にあるように（modelに食わせる直前で）

boxes.append(torch.tensor([[0, 0, 0, 0]]).to(self.device).float())

でサイズ1のbboxを渡してるんですけど、ちょっとイレギュラーなことしている感じあるので大丈夫か確認したい torch.zeros(0, 4)みたいにサイズ0のbboxを渡すとエラーが起きてしまった

Toma0916 commented 3 years ago 
boxes.append(torch.tensor([[0, 0, 0, 0]]).to(self.device).float())

これは合ってそう？（モデルの使い方にbb無い時そうしろってあったよね確か）