神操作!只需5行代码,就能做出一个图像识别AI
首先,你必须通过此链接下载IdenProf数据集的zip。 你还可以在下面的链接中的IdenProf GitHub存储库中查看经过培训以识别职业的人工智能模型的所有详细信息和示例结果。 https://github.com/OlafenwaMoses/IdenProf 因为训练人工智能模型需要高性能的计算机系统,所以我强烈建议你确保要用于此训练的计算机/笔记本电脑具有NVIDIA GPU。 另外,如果你提供免费的NVIDIA K80 GPU进行实验,则可以使用Google Colab进行此实验。 然后,你必须安装ImageAI及其依赖项。 1)从Python语言官方网站下载并安装Python 3 https://python.org 2)通过pip安装以下pip: i. TensorFlow pip3 install tensorflow ii. OpenCV pip3 install opencv-python iii. Keras pip3 install keras iv. ImageAI pip3 install imageai --upgrade 用你想要给它的任何名称创建一个python文件,例如“ FirstTraining.py”。 将IdenProf数据集的zip复制到Python文件所在的文件夹中。 然后将其解压缩到同一文件夹中。 ·然后将下面的代码复制到python文件(例如FirstTraining.py)中。 from imageai.Prediction.Custom import ModelTraining model_trainer = ModelTraining() model_trainer.setModelTypeAsResNet() model_trainer.setDataDirectory("idenprof") (编辑:晋中站长网) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |