pytorch-image-classifier-template. This code modifies the output layer of image classification network commonly used in pytorch. The modified model can be used to process any number of image classification data. At the same time, the pre-training parameters and re-training parameters are distinguished for fine-tuning and component training of parameters.

github.com/xxcheng0708/pytorch-image-classifier-template

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.