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