ChessAnalysisPipeline 0.0.2__py3-none-any.whl → 0.0.4__py3-none-any.whl

CHAP/inference/processor.py

@@ -0,0 +1,68 @@
+#!/usr/bin/env python                                                                                                       
+#-*- coding: utf-8 -*-                                                                                                      
+#pylint: disable=                                                                                                           
+'''                                                                                                                         
+File       : processor.py                                                                                                   
+Author     : Valentin Kuznetsov <vkuznet AT gmail dot com>                                                                  
+Description: Processor module                                                                                               
+'''
+
+# system modules
+from time import time
+
+# local modules
+from CHAP import Processor
+
+class TFaaSImageProcessor(Processor):
+    '''
+    A Processor to get predictions from TFaaS inference server.
+    '''
+    def process(self, data, url, model, verbose=False):
+        '''
+        process data API
+        '''
+
+        t0 = time()
+        self.logger.info(f'Executing "process" with url {url} model {model}')
+
+        data = self._process(data, url, model, verbose)
+
+        self.logger.info(f'Finished "process" in {time()-t0:.3f} seconds\n')
+
+        return(data)
+
+    def _process(self, data, url, model, verbose):
+        '''Print and return the input data.
+
+        :param data: Input image data, either file name or actual image data
+        :type data: object
+        :return: `data`
+        :rtype: object
+        '''
+
+        from MLaaS.tfaas_client import predictImage
+        from pathlib import Path
+
+        self.logger.info(f'input data {type(data)}')
+        if isinstance(data, str) and Path(data).is_file():
+            imgFile = data
+            data = predictImage(url, imgFile, model, verbose)
+        else:
+            rdict = data[0]
+            import requests
+            img = rdict['data']
+            session = requests.Session()
+            rurl = url + '/predict/image'
+            payload = dict(model=model)
+            files = dict(image=img)
+            self.logger.info(f'HTTP request {rurl} with image file and {payload} payload')
+            req = session.post(rurl, files=files, data=payload )
+            data = req.content
+            data = data.decode('utf-8').replace('\n', '')
+            self.logger.info(f'HTTP response {data}')
+
+        return(data)
+
+if __name__ == '__main__':
+    from CHAP.processor import main
+    main()

CHAP/inference/reader.py

@@ -0,0 +1,5 @@
+#!/usr/bin/env python  
+
+if __name__ == '__main__':
+    from CHAP.reader import main
+    main()

CHAP/inference/writer.py

@@ -0,0 +1,5 @@
+#!/usr/bin/env python  
+
+if __name__ == '__main__':
+    from CHAP.writer import main
+    main()

CHAP/pipeline.py

@@ -50,7 +50,7 @@ class Pipeline():
                 self.logger.info(f'Calling "write" on {item}')
                 data = item.write(data, **kwargs)
 
-        self.logger.info(f'Exectuted "exectute" in {time()-t0:.3f} seconds')
+        self.logger.info(f'Executed "execute" in {time()-t0:.3f} seconds')
 
 class PipelineObject():
     """