Skip to content

Pretrained Weights

Latest
Compare
Choose a tag to compare
@alcros33 alcros33 released this 24 Mar 21:14
· 1 commit to master since this release

This release contains the pre-trained weights for all the models used in this repo.

  • colorizer.pkl is the fastai learner containing the main colorizer model. It should be placed in the models/ folder.
  • xyv_vgg.pth is the custom vgg-net used for perceptual loss. It should be placed in the models/ folder.
  • anime2sketch.pth is the anime2sketch model. It should be placed in the LA/models/ folder.
  • sketch_simplification.pth is the sketch_simplification model. It should be placed in the LA/models/ folder.
  • sketchKeras.pth is the sketchKeras model. It should be placed in the LA/models/ folder.