Project from pyimagesearch, identify both cloth type and cloth color
From this project, you can learn how to:
- Create your own dataset and feed it to dataloader in pytorch
- Define your model based on pre-trained model (resnet18) and fine-tune it
- Save the model
- Draw model graph using tensorboardX