# 定义函数用于将.csv中存储的像素点label载入程序,整体读成列表字典,图片名称与类别作为key,
# 像素坐标列表作为value
def readDictCSV(fileName):
dataDict = {}
with open(fileName, "r") as csvFile:
data = csv.DictReader(csvFile)
for row in data:
dataDict.setdefault(row['ImageId_ClassId'],[])
.append(
row['EncodedPixels'].split(' ',len(row['EncodedPixels'])//2-1
))
csvFile.close()
return dataDict
if __name__ == '__main__':
filename = '/home/dell-g3/PycharmProjects/pytorch-project/Image_Segmentation-master/data/train.csv'
name = readDictCSV(filename)
lll = []
for i in range(4):
l = name['0002cc93b.jpg'+'_'+str(i+1)]
if l == [['']]:
ll = np.array([[-1,-1]]) # 这里有致命错误,tenser在stack时要保证大小完全相同,列表可以实现但张量不行
else:
ll = np.array(l).astype(int).reshape(-1, 2)
ll = ll.tolist()
lll.append(ll)
LLL = torch.Tensor(lll)
最后
以上就是坚强滑板最近收集整理的关于label的输出形式,折腾出来才发现太复杂了的全部内容,更多相关label内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复