KichangKim / DeepDanbooru

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

Load weights in Pytorch #35

Closed ghost closed 3 years ago

ghost commented 3 years ago

Is it possible to convert the weights for Pytorch?

KichangKim commented 3 years ago

You can convert DeepDanbooru model to ONNX format by using 3rdparty library and also import ONNX to pytorch too. But I can't sure because I have never used such libraries.

https://github.com/onnx/tensorflow-onnx https://github.com/pytorch/pytorch/issues/21683

ghost commented 3 years ago

For some reason I couldn't install tf. Could you release the model in ONNX format? Thanks!!

KichangKim commented 3 years ago

Unfortunately I have no time for DeepDanbooru now. I think these sites will help you:

https://github.com/ysh329/deep-learning-model-convertor https://github.com/halcy/DeepDanbooruActivationMaps