python cnn图像分类_CNN图像分类

Tensorflow提供的图像分类工具

Tensorflow提供了一个能基于预训练模型进行重训练的脚本：https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py，只需提供图像即可完成训练，不用写代码，预训练模型可以使用inception v3，或者准确率相对较低但运行更快的mobilenet。

目前这个脚本已经有了针对tensorflow 2.0的更新版本：make_image_classifier.py

通过以下命令即可完成在几分钟内完成训练：

$ make_image_classifier \

--image_dir my_image dir \

--tfhub_module https://tfhub.dev/google/tf2-preview/mobilenet_v2/feature_vector/4 \

--image_size 224 \

--saved_model_dir my_dir/new_model \

--labels_output_file class_labels.txt \

--tflite_output_file new_mobile_model.tflite

下面讲下参数。

--image_dir，存放训练图像的目录，结构如下：

my image_dir

|-- cat

| |-- a_feline_photo.jpg

| |-- another_cat_pic.jpg

| `-- ...

|-- dog

| |-- PuppyInBasket.JPG

| |