物体検出で個数がカウントされない

物体検出数のカウント出力を試みています。
物体検出数のカウント出力の項目でつまづいてしまいました。

エラーは出ていないのですが、個数がカウントされません。
どうすればいいかご教授いただければ幸いです。

detect_2.pyのダウンロード先

追加したコード↓

# Write results for *xyxy, conf, cls in reversed(det): if save_txt: # Write to file xywh = (xyxy2xywh(torch.tensor(xyxy).view(1, 4)) / gn).view(-1).tolist() # normalized xywh line = (cls, *xywh, conf) if save_conf else (cls, *xywh) # label format with open(f'{txt_path}.txt', 'a') as f: f.write(('%g ' * len(line)).rstrip() % line + '\n') if save_img or save_crop or view_img: # Add bbox to image c = int(cls) # integer class label = None if hide_labels else (names[c] if hide_conf else f'{names[c]} {conf:.2f}') annotator.box_label(xyxy, label, color=colors(c, True)) if save_crop: save_one_box(xyxy, imc, file=save_dir / 'crops' / names[c] / f'{p.stem}.jpg', BGR=True) # 追加したコードPrint counter n_1 = (det[:, -1] == 1).sum() #ラベルAの総数をカウント a = f"{n_1} "#{'A'}{'s' * (n_1 > 1)}, " cv2.putText(im0, "cabbage counter " , (20, 50), 0, 1.0, (71, 99, 255), 3) cv2.putText(im0, "A : " + str(a), (20, 100), 0, 1.5, (71, 99, 255), 3) n_2 = (det[:, -1] == 1).sum() #ラベルBの総数をカウント b = f"{n_2} "#{'A'}{'s' * (n_1 > 1)}, " cv2.putText(im0, "B : " + str(b), (20, 150), 0, 1.5, (0, 215, 255), 3) n_3 = (det[:, -1] == 2).sum() #ラベルCの総数をカウント d = f"{n_3} "#{'A'}{'s' * (n_1 > 1)}, " cv2.putText(im0, "C : " + str(d), (20, 200), 0, 1.5, (154, 250, 0), 3)

実行結果↓
イメージ説明

コメントを投稿

0 コメント