一个小小的图像分类库。
neuron-ml的Python项目详细描述
神经元
一个小型的、高层次的图像分类转换学习库神经元到底是什么?
neuron是一个小型的库,旨在简化图像分类(如果你不知道什么是图像分类,它是一个从图像中分辨出图像上的对象/事物/特征的过程)。
使用neuron,您将能够在5行代码下构建生产级模型。是的,你读对了:5行。在像TensorFlow、Pythorch或Keras这样的通用机器学习库中,您可以在数百行中完成它。
Of course, these libraries are much more complex and versatile than Neuron. Neuron isn't replacing these libraries if you need to build your own graph, but if you're doing so, you probably already know this.
安装
复制-粘贴到终端中
pip install neuron-ml
演示
下面是神经元能做的一个例子:
importneuron_mlasn# TensorFlowdata=n.load("./dataset/")# formats the datamodel=n.train(data)# train the datan.export(model,["./Model.pb","./Labels.txt"])# Exports everythingn.clean(model)# Clean temporary files
它还可以加载文件并对其进行分类(在将其用于生产之前,请确保您有良好的硬件,因为模型可能需要5秒的时间来加载和运行)。
importneuron_mlasnmodel=n.model(["./Model.pb","./Labels.txt"])# Load the modelgraph=n.graph(model)# Generate the graphlabels=n.labels(model)# Get the labelsn.classify(graph,labels,"./dataset/Celery/celery-1.jpg")# Classify. Will return a result object
See the wiki for more informations.
版本控制
我们使用SemVer进行版本控制。有关可用的版本,请参见tags on this repository。
作者
- arthur guiot-initial work-@arguiot
还要看看参与这个项目的contributors的列表。如果你不编码,但你有很好的想法,不要犹豫写你的想法在问题部分。如果你的想法被接受,我会把你列入这个名单。
许可证
此项目是根据麻省理工学院许可证授权的-有关详细信息,请参见<;许可证>;文件
版权所有©2018 Arthur Guiot保留所有权利。