File size: 1,367 Bytes
3fd659d bf871d0 3fd659d bf871d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
---
tags:
- pytorch
- vision
- face-analysis
- beauty-prediction
- scut-fbp5500
library_name: generic
license: mit
---
# SCUT-FBP5500 ResNet-18 Model
This repository contains the PyTorch weights for a facial beauty prediction model trained on the [SCUT-FBP5500 dataset](https://github.com/HCIILAB/SCUT-FBP5500-Database-Release).
## Model Files
- **`resnet18.pth`**: Original weights (saved in Python 2). Requires `encoding='latin1'` to load in Python 3.
- **`resnet18_py3.pth`**: Converted weights compatible with Python 3+ (Recommended).
## Model Details
- **Architecture:** ResNet-18
- **Dataset:** SCUT-FBP5500 (Facial Beauty Prediction)
- **Framework:** PyTorch
## Origin
The original weights were provided by [brian-xu](https://github.com/brian-xu/torch-converter). The `resnet18_py3.pth` file is a converted version using `_use_new_zipfile_serialization=True`.
## Usage (Python 3)
```python
import torch
import torchvision.models as models
model = models.resnet18(num_classes=1)
# Load the converted weights (Recommended)
state_dict = torch.load("resnet18_py3.pth")
model.load_state_dict(state_dict)
model.eval()
```
## Licensing
Please refer to the original [SCUT-FBP5500 Database](https://github.com/HCIILAB/SCUT-FBP5500-Database-Release) for usage terms regarding the dataset and derived models.
|