mirror of
https://gitee.com/mymagicpower/AIAS.git
synced 2024-11-29 18:58:16 +08:00
.. | ||
doc/img | ||
lib | ||
src | ||
classification_imagenet_sdk.iml | ||
pom.xml | ||
README.md |
目录:
图片分类(支持imagenet数据集分类)SDK
识别图片1000种分类。
支持分类如下:
- 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包含两个分类器:
- Darknet53Classification
backbone: darknet53, dataset:imagenet - MobilenetClassification
backbone: Mobilenetv3_small, dataset:imagenet
运行例子 - Darknet53ClassificationExample
运行成功后,命令行应该看到下面的信息:
...
n03063599 coffee mug : 0.39411163330078125
[INFO ] - [
class: "n03063599 coffee mug", probability: 0.39411
class: "n07930864 cup", probability: 0.21575
class: "n07920052 espresso", probability: 0.03084
class: "n07932039 eggnog", probability: 0.02415
class: "n04579145 whiskey jug", probability: 0.01683
]
帮助
添加依赖库:lib/aias-classification-imagenet-lib-0.1.0.jar