基于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 ·
1983 阅读 ·
10 评论