mirror of
https://gitee.com/mymagicpower/AIAS.git
synced 2024-12-02 12:17:37 +08:00
.. | ||
build/output | ||
doc/img | ||
src/main | ||
biggan_sdk.iml | ||
pom.xml | ||
README.md |
目录:
BIGGAN 图像自动生成SDK
能够自动生成1000种类别(支持imagenet数据集分类)的图片。
支持分类如下:
- tench, Tinca tinca
- goldfish, Carassius auratus
- great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias
- tiger shark, Galeocerdo cuvieri
- hammerhead, hammerhead shark
- electric ray, crampfish, numbfish, torpedo
- stingray
- cock
- hen
- ostrich, Struthio camelus
- brambling, Fringilla montifringilla
- goldfinch, Carduelis carduelis
- house finch, linnet, Carpodacus mexicanus
- junco, snowbird
- indigo bunting, indigo finch, indigo bird, Passerina cyanea
- robin, American robin, Turdus migratorius
- ...
SDK包含两个分类器:
size 支持 128, 256, 512三种图片尺寸 如:size = 512; imageClass 支持imagenet类别0~999 如:imageClass = 156;
运行例子 - BigGAN
运行成功后,命令行应该看到下面的信息:
...
[INFO ] - Number of inter-op threads is 4
[INFO ] - Number of intra-op threads is 4
[INFO ] - Generated image has been saved in: build/output/
Pytorch模型代码: https://github.com/mymagicpower/BigGAN-Generator-Pretrained-Pytorch