test/AIAS
1
0
Fork 0
mirror of https://gitee.com/mymagicpower/AIAS.git synced 2024-12-02 04:08:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
725a571be8
AIAS/classification_imagenet_sdk
History
Calvin b14782b3a9
Update README.md
2021-07-11 20:28:20 +08:00
..
doc/img Add imagenet classification. 2021-07-11 20:07:50 +08:00
lib Add imagenet classification. 2021-07-11 20:07:50 +08:00
src Add imagenet classification. 2021-07-11 20:07:50 +08:00
classification_imagenet_sdk.iml Add imagenet classification. 2021-07-11 20:07:50 +08:00
pom.xml Add imagenet classification. 2021-07-11 20:07:50 +08:00
README.md Update README.md 2021-07-11 20:28:20 +08:00

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

  • 测试图片 img

运行成功后，命令行应该看到下面的信息:

...
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