5bb91bfefafbf9cc643aa1a3c7c4d8fdfec46e56,ml_tools/eolearn/ml_tools/classifier.py,ImageClassificationMaskTask,execute,#ImageClassificationMaskTask#Any#,477

Before Change

        :return: Outputs EOPatch with n classification masks appended to out_feature_type with out_feature_name key
        :rtype: EOPatch
        
        array = eopatch.get_feature(self.in_feature_type, self.in_feature_name)

        clf_mask = self.classifier.image_predict(array)

        eopatch.add_feature(self.out_feature_type, self.out_feature_name, clf_mask)

        return eopatch

After Change

        in_type, in_name = next(self.input_feature(eopatch))
        out_type, out_name = next(self.input_feature())

        eopatch[out_type][out_name] = self.classifier.image_predict(eopatch[in_type][in_name])

        return eopatch

Instances