物体検出数のカウント出力を試みています。
物体検出数のカウント出力の項目でつまづいてしまいました。
エラーは出ていないのですが、個数がカウントされません。
どうすればいいかご教授いただければ幸いです。
追加したコード↓
# 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] == 0).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)
printしたコード↓
# Print results t = tuple(x.t / seen * 1E3 for x in dt) # speeds per image LOGGER.info(f'Speed: %.1fms pre-process, %.1fms inference, %.1fms NMS per image at shape {(1, 3, *imgsz)}' % t) if save_txt or save_img: s = f"\n{len(list(save_dir.glob('labels/*.txt')))} labels saved to {save_dir / 'labels'}" if save_txt else '' LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}{s}") if update: strip_optimizer(weights[0]) # update model (to fix SourceChangeWarning) #print print(FILE, ROOT, source, save_img, is_file, is_url, webcam, save_dir, device, model, stride, names, pt, imgsz, view_img, dataset, bs, dataset ,dataset, vid_path, vid_writer, seen, windows, dt, im, visualize, pred, p, save_path, txt_path, gn, imc, annotator, xywh, line, c, label, n_1, a , cv2.putText, cv2.putText, n_2, b, cv2.putText, n_3, d, cv2.putText, im0, dataset.mode, t, s)
実行結果↓
