The microocr.net from williamlzw

microocr.net's People

Contributors

Stargazers

Watchers

microocr.net's Issues

The

Do I need to modify the some parameters to recognize 10 to 20 characters?

I found that in this project, the accuracy of identifying 1 to 4 characters at one time is relatively high. If the number of characters increases, no matter how you train, the output results will have more problems.

I tried to replace the Decoder method, and the reasoning speed seems to be faster

  public List<Tuple<string, List<float>>> Decode(torch.Tensor preds, bool raw = false)
        {
            preds = torch.nn.functional.softmax(preds, 2);
            var ret = preds.max(2);
            var preds_idx = ret.indexes.detach().to_type(ScalarType.Int32).to(CPU);
            var preds_score = ret.values.detach().to(CPU);
            List<Tuple<string, List<float>>> retList = new List<Tuple<string, List<float>>>();
            for (int i = 0; i < preds_idx.shape[0]; i++)
            {
                var word = preds_idx[i].data<int>().ToArray();
                if (raw)
                {
                    var scoreList = preds_score[i].data<float>().ToArray();
                    var score = scoreList[0];
                    string retStr = "";
                    List<float> retScoreList = new List<float>();
                    for (int j = 0; j < word.Length; j++)
                    {
                        var charIdx = word[j];
                        retStr = retStr + _idx2char[charIdx];
                    }
                    retScoreList.Add(score);
                    var tuple = new Tuple<string, List<float>>(retStr, retScoreList);
                    retList.Add(tuple);
                }
                else
                {
                    var scoreList = preds_score[i].data<float>().ToArray();
                    string retStr = "";
                    List<float> retScoreList = new List<float>();
                    for (int j = 0; j < word.Length; j++)
                    {
                        var charIdx = word[j];
                        if (charIdx != 0 && !(j > 0 && (word[j - 1]) == charIdx))
                        {
                            var bb = charIdx;
                            retStr = retStr + _idx2char[bb];
                            var score = scoreList[j];
                            retScoreList.Add(score);
                        }
                    }
                    var tuple = new Tuple<string, List<float>>(retStr, retScoreList);
                    retList.Add(tuple);
                }
            }
            return retList;
        }