基于pytorch的目标检测数据增强(tensor数据流版本)
我的示例代码的dataloader中打包传入的是一个target字典,里面包括boxes和label,如果你们传入的是boxes和label,直接修改参数就行了,然后因为我传入的image和target都是经过torch的转换,数据格式是tensor,所以有一些转换格式的代码,然后图片shape是(c,h,w),随机概率设的是0.3,都按需要修改就行。
随机缩放
class randomScale(object):
def __call__(self,image,target):
#
原创
2021-02-01 11:31:25 ·
1942 阅读 ·
10 评论