分类 python 下的文章 - 白帽Wiki

2021-03-05 huoji 3 条评论

[2021]一些常见的机器视觉用到的神经网络结构(keras)

收藏一些常见的机器视觉用到的神经网络结构,以后不用自己构造网络了!
这些都咋用呢? 我已经写好祖传代码了:
```python
def build_save_model(train_data, test_data):
    # 构建模型 CBAPD
    network_model = VGG16()

network_model.compile(optimizer='adam',
                          loss=tensorflow.keras.losses.SparseCategoricalCrossentropy(
                              from_logits=False),
                          metrics=['accuracy'])
    #cp_callback = tensorflow.keras.callbacks.ModelCheckpoint(filepath=checkpoint_save_path,save_weights_only=True,save_best_only=True)
    # 如果要保存点自己在fit加这个参数
    # callbacks=[cp_callback]
    #checkpoint_save_path = "./checkpoint.ckpt"
    # if os.path.exists(checkpoint_save_path + '.index'):
    #    print('-------------load the model-----------------')
    #    network_model.load_weights(checkpoint_save_path)
    network_model.fit(train_data, epochs=100,
                      validation_data=test_data, batch_size=64)
    network_model.summary()
    network_model.evaluate(test_data)
    tensorflow.saved_model.save(
        network_model, '\\model\\')
```
VGG网络:

阅读全文

2021-02-24 huoji 1 条评论