ethan-sui

> Traceback (most recent call last): > File "train.py", line 128, in > for batch_i, (_, imgs, targets) in enumerate(dataloader): > File "/root/anaconda3/envs/pytorch/lib/python3.6/site-packages/torch/utils/data/dataloader.py", line 346, in **next** > data =...

> 去掉水平反转增强试试 > aug = A.Compose([ > # A.HorizontalFlip(p=0.5), > A.HueSaturationValue(hue_shift_limit=10, sat_shift_limit=10, val_shift_limit=10, p=0.5), > # A.ShiftScaleRotate(shift_limit=0.1, scale_limit=0.1, rotate_limit=5, border_mode=0, p=0.5) > ], bbox_params={'format':'coco', 'label_fields': ['category_id']}) 方便加个好友吗，QQ：2803788198