rle_ensemble_loader.py 文件源码

python
阅读 25 收藏 0 点赞 0 评论 0

项目:carvana-challenge 作者: chplushsieh 项目源码 文件源码
def __getitem__(self, idx):

        img_name = self.img_names[idx]

        ensembled_mask = np.zeros((const.img_size[0], const.img_size[1]))

        for i, submission in enumerate(self.submissions):

            rle = submission[img_name]
            mask = run_length.decode(rle)
            weighted_mask = np.multiply(mask, self.weights[i])
            ensembled_mask = np.add(ensembled_mask, weighted_mask)

        ensembled_mask[ ensembled_mask > 0.5 ] = 1
        ensembled_mask[ ensembled_mask <= 0.5 ] = 0

        # plt.imshow(ensembled_mask)
        # plt.show()

        ensembled_rle = run_length.encode(ensembled_mask)

        return img_name, ensembled_rle
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号