Use ImageNet pretrained classification models to predict the labels of the AFHQ dataset.
python predict.py --model resnet18 --dataset_path /path/to/afhq --output_path labels.csv- The dataset can be downloaded from here.
-
Notifications
You must be signed in to change notification settings - Fork 0
songquanpeng/AFHQ-label
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Use ImageNet pretrained classification models to predict the labels of the AFHQ dataset.
Topics
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published