Deep Extraction of Manga Structural Lines in python
The (Official) PyTorch Implementation of the paper “Deep Extraction of Manga Structural Lines”
Usage
model_torch.py [source folder] [output folder]
Example:
model_torch.py ./pytorchTestCases/ ./pytorchResults/
The model weights (erika.pth)
Please refer to the release section of this repo. Alternatively, you may use this link:
https://www.dropbox.com/s/y8pulix3zs73y62/erika.pth?dl=0
Requirement
- Python3
- PyTorch (tested on version 1.9)
- Python-opencv
How the model is prepared
The PyTorch weights are exactly the same as the theano(!) model. I make some efforts to convert the original weights to the new model and ensure the overall error is less than 1e-3 over the image range from 0-255.
Moreover, the functional PyTorch interface allows easier fine-tuning of this model. You can also take the whole model as a sub-module for your own work (e.g., use the on-the-fly extraction of lines as a structural constraint).
About model