分类：加载batch时卡死

2021
07-22

Pytorch dataloader在加载最后一个batch时卡死的解决

问题：自己写了个dataloader，为了部署方便，用OpenCV的接口进行数据读取，而没有用PIL，代码大致如下：def__getitem__(self,idx):sample=self.samples[idx]img=cv2.imread(sample[0])img=cv2.resize(img,tuple(self.input_size))img=cv2.cvtColor(img,cv2.COLOR_BGR2RGB)#ifnotself.valandrandom.randint(1,10)<3:#img=self.img_aug(img)... 继续阅读 >