search

KichangKim / DeepDanbooru

AI based multi-label girl image classification system, implemented by using TensorFlow.
MIT License
2.65k stars 260 forks source link

accept BytesIO #9

Closed rachmadaniHaryono closed 4 years ago

rachmadaniHaryono commented 4 years ago

evaluate_imagenow can accept image BytesIO instead just image filename.

this is needed when the function is used on flask views. in that case the received posted image can be directly processed instead saving it to disk and get image filename

this include test and type hint for both affected function

KichangKim commented 4 years ago

Event six package is automatically installed with tensorflow, but I think that it should be included DeepDanbooru's dependencies if DD use it explicitly.