iOS 通過(guò)VisionKit從PDF中提取TXT

懶惰與近視是程序員進(jìn)步的第一生產(chǎn)力(他絕不會(huì)承認(rèn)是因?yàn)樨毟F)。

場(chǎng)景:小A想要展示一本PDF文檔,通過(guò)WKWebView直接展示是首選方案,因?yàn)楹?jiǎn)單方便。但是,因?yàn)樯頌槌绦騿T的驕傲,促使他使用了更高級(jí)的方式。
于是,他這樣做:

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    [self.view addSubview:self.pdfView];
}

- (PDFView *)pdfView {
    if (!_pdfView) {
        _pdfView = [[PDFView alloc] initWithFrame:self.view.bounds];
        _pdfView.document = self.document;
        _pdfView.delegate = self;
    }
    return _pdfView;
}

- (PDFDocument *)document {
    if (!_document) {
        NSURL *url = [[NSBundle mainBundle] URLForResource:@"xxx" withExtension:@"pdf"];
        _document = [[PDFDocument alloc] initWithURL:url];
    }
    return _document;
}

一切都那么順利,他展示出了pdf文件每一頁(yè)的內(nèi)容。但是人的欲望總是難以得到滿足,雖然展示出來(lái)了數(shù)據(jù),但是由于只是單純的展示PDF文件,在小小的iPhone上面,什么都顯得那么小,對(duì)于近視的小A來(lái)說(shuō),真的是螞蟻爬爬。小A不禁想到,如果可以使用百度小說(shuō)的閱讀器直接打開,那么就可以隨心所欲的改變字體大小以及排版結(jié)構(gòu)了,近不近視的完全不受影響嘛。

所幸,PDFKit可以讓用戶獲取到每一頁(yè)的page內(nèi)容,然后通過(guò)page可以拿到內(nèi)容所展示的文字,那么,就可以將每一頁(yè)的內(nèi)容進(jìn)行整合展示成txt文件,用任意你想用的程序進(jìn)行展示,做法也非常簡(jiǎn)單:

for (int i = 0; i < [self.document pageCount]; i++) {
        PDFPage *page = [self.document pageAtIndex:i];
        NSLog(@"--!!!-- content: %@", page.string);
//        NSLog(@"--!!!-- content: %@", page.attributedString);
    }

小A很開心,因?yàn)樗梢院苁孢m的瀏覽他想要看到的內(nèi)容了,他不禁想到,近視真的是人類進(jìn)步的階梯。他一直這樣開心,直到他看到文章出現(xiàn)了斷層。。

WTF??什么情況,正在生離死別的緊要關(guān)頭怎么就開始你儂我儂了?

于是,小A開始了苦逼的數(shù)據(jù)校驗(yàn),試圖找到那只該死的『蟲子』。幸運(yùn)的是,原文件支持他很快的找到了缺失的那一章節(jié),原來(lái)那一頁(yè)P(yáng)DF不是簡(jiǎn)單text,而是令人崩潰的image。。圖片轉(zhuǎn)文字,需要用的傳說(shuō)中的OCR技術(shù),想到一系列的算法識(shí)別以及數(shù)據(jù)喂養(yǎng),小A頭大如斯,從現(xiàn)場(chǎng)景來(lái)看,通過(guò)接入各大廠商的OCR收費(fèi)sdk是首選方案,但是,貧窮的程序員怎么可以忍受嗟來(lái)之食?于是,他開始關(guān)注相關(guān)PDF轉(zhuǎn)TXT的APP,果然天下沒(méi)有白吃的午餐,想要不花錢就進(jìn)行轉(zhuǎn)換,就有了轉(zhuǎn)換內(nèi)容大小的限制,一次只能轉(zhuǎn)化1M的內(nèi)容,小A望著47.5M的文件陷入了沉思。。

最終,小A將目光投向了VisionKit,謝天謝地,強(qiáng)大的VisionKit可以直接進(jìn)行中文識(shí)別,貧窮的小A覺(jué)得一切都是最好的安排。于是誕生了下面的一段代碼:

        PDFPage *page = [self.document pageAtIndex:i];
        UIImage *image = [page thumbnailOfSize:[page boundsForBox:kPDFDisplayBoxMediaBox].size forBox:kPDFDisplayBoxMediaBox];
        CGImageRef img = image.CGImage;
        
        VNImageRequestHandler *handler = [[VNImageRequestHandler alloc] initWithCGImage:img options:nil];
        VNRecognizeTextRequest *textRequest = [[VNRecognizeTextRequest alloc] initWithCompletionHandler:^(VNRequest * _Nonnull request, NSError * _Nullable error) {
            
            if (error) {
                NSLog(@"--!!!-- error : %@", error);
            } else {
                NSArray<VNRecognizedTextObservation *> *result = request.results;
                // 處理結(jié)果
            }
        }];
        textRequest.recognitionLevel = VNRequestTextRecognitionLevelAccurate;
        textRequest.usesLanguageCorrection = NO;
        textRequest.recognitionLanguages = @[@"zh-Hans", @"en-US"];
//        textRequest.revision = VNRecognizeTextRequestRevision2;
        
        [handler performRequests:@[textRequest] error:nil];

在應(yīng)用過(guò)程中,小A發(fā)現(xiàn)可以通過(guò)recognitionLevel屬性來(lái)選擇識(shí)別的級(jí)別,是快速識(shí)別或是精準(zhǔn)識(shí)別。usesLanguageCorrection用來(lái)確認(rèn)是否需要在識(shí)別過(guò)程中使用語(yǔ)言更正,如果關(guān)閉此選項(xiàng),對(duì)性能有較大的正向影響,但可能會(huì)影響識(shí)別結(jié)果的正確性。但是基本已經(jīng)不影響閱讀了。

至此,小A終于可以開開心心地隨意閱讀他想要閱讀的pdf文件了。

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容