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.

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

Added warning message when --single-cls is used incorrectly by [#13028](https://github.com/ultralytics/yolov5/issues/13028) #13032

Open
wants to merge 26 commits into
base: master
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
025694f
Added warning message when --single-cls is used incorrectly
Le0v1n May 21, 2024
e753411
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant May 24, 2024
c2b7d72
Trigger retest
Le0v1n May 24, 2024
5bd1476
re-CI
Le0v1n May 24, 2024
522b3ed
Merge branch 'add-warning-single-cls' of https://github.com/Le0v1n/yo…
Le0v1n May 24, 2024
227789d
Auto-format by https://ultralytics.com/actions
UltralyticsAssistant May 24, 2024
f505e63
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant May 28, 2024
305f529
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant May 29, 2024
1fcf888
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant May 29, 2024
5611c7f
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant May 30, 2024
b1e0213
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 8, 2024
aa80706
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 8, 2024
c641d26
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 9, 2024
9d9a0f8
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 16, 2024
86b79bd
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 16, 2024
33d9239
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 17, 2024
89ff432
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 19, 2024
40d09e9
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 19, 2024
fe8dbbe
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 20, 2024
81e2d00
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 20, 2024
a5c581d
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 20, 2024
4b25730
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 20, 2024
c6eed0d
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 22, 2024
7a90cc5
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 30, 2024
c1061b3
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 30, 2024
cc535dd
Merge branch 'master' into add-warning-single-cls
UltralyticsAssistant Jun 30, 2024
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
9 changes: 8 additions & 1 deletion train.py
Original file line number Diff line number Diff line change
Expand Up @@ -176,7 +176,14 @@ def train(hyp, opt, device, callbacks):
data_dict = data_dict or check_dataset(data) # check if None
train_path, val_path = data_dict["train"], data_dict["val"]
nc = 1 if single_cls else int(data_dict["nc"]) # number of classes
names = {0: "item"} if single_cls and len(data_dict["names"]) != 1 else data_dict["names"] # class names
if single_cls and len(data_dict["names"]) != 1:
LOGGER.warning(
"WARNING ⚠️ Please check the dataset to ensure that when --single-cls is enabled, "
"the number of classes in the dataset is 1."
)
names = {0: "item"} # combine all classes into a single class
else:
names = data_dict["names"] # class names
is_coco = isinstance(val_path, str) and val_path.endswith("coco/val2017.txt") # COCO dataset

# Model
Expand Down
Loading