yolov5/v7 单类训练

我认为单类训练主要是用来区分前景和背景。

实际做的时候：加一个参数就行 --single-cls

nohup python -m torch.distributed.launch --nproc_per_node 2 --master_port 9527 train.py --workers 16 --device 0,1 --sync-bn --batch-size 128 --single-cls &

官方教程

if single_cls: # 单类训练
for x in self.labels: # self.labels(样本数，5(cls,xywh))
x[:, 0] = 0 # 让cls 变成0