Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add confidence score to label txt files #994

Merged
merged 9 commits into from
Oct 20, 2020
Merged

Add confidence score to label txt files #994

Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
e37f51d
add contrast to conf/class against bbox rectangle color
oakhtar147 Sep 18, 2020
bc16152
added feature to save conf in txt labels
oakhtar147 Sep 18, 2020
5c98152
Update general.py
glenn-jocher Oct 20, 2020
f81d5c3
Update detect.py
glenn-jocher Oct 20, 2020
ae9c5d4
Update detect.py
glenn-jocher Oct 20, 2020
7aa2dde
Merge branch 'master' into add-conf_score-to-label
glenn-jocher Oct 20, 2020
1ff288e
Update detect.py
glenn-jocher Oct 20, 2020
e54303c
Update detect.py
glenn-jocher Oct 20, 2020
5be9332
Update detect.py
glenn-jocher Oct 20, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 3 additions & 1 deletion detect.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,9 @@ def detect(save_img=False):
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, conf, *xywh) if opt.save_conf else (cls, *xywh) # label format
with open(txt_path + '.txt', 'a') as f:
f.write(('%g ' * 5 + '\n') % (cls, *xywh)) # label format
f.write(('%g ' * len(line) + '\n') % line)

if save_img or view_img: # Add bbox to image
label = '%s %.2f' % (names[int(cls)], conf)
Expand Down Expand Up @@ -154,6 +155,7 @@ def detect(save_img=False):
parser.add_argument('--device', default='', help='cuda device, i.e. 0 or 0,1,2,3 or cpu')
parser.add_argument('--view-img', action='store_true', help='display results')
parser.add_argument('--save-txt', action='store_true', help='save results to *.txt')
parser.add_argument('--save-conf', action='store_true', help='output confidences in --save-txt labels')
parser.add_argument('--classes', nargs='+', type=int, help='filter by class: --class 0, or --class 0 2 3')
parser.add_argument('--agnostic-nms', action='store_true', help='class-agnostic NMS')
parser.add_argument('--augment', action='store_true', help='augmented inference')
Expand Down