CV之NS之VGG16：基于預訓練模型VGG16訓練COCO的train2014數據集實現訓練《神奈川沖浪里》風格配置yml文件

?

?

目錄

一、訓練

1、《神奈川沖浪里》風格

2、配置yml文件

二、TensorBoard中監控訓練情況

---

?

?

?

一、訓練

1、《神奈川沖浪里》風格

風格如圖

2、配置yml文件

?

style_image: img/wave.jpg # targeted style image指定原始風格圖像 ##naming、model_path 兩個量定義了最終的checkpoint 和監控信息。events文件會被保存在models/wave文件夾下 naming: "wave" # the name of this model一般和圖像名字保持一致. Determine the path to save checkpoint and events file. model_path: models # root path根目錄 to save checkpoint and events file. The final path would be <model_path>/<naming>## Weight of the loss各個損失的權重 content_weight: 1.0 # weight for content features loss內容損失權重 style_weight: 220.0 # weight for style features loss風格損失權重 tv_weight: 0.0 # weight for total variation loss，(1)在本項目中，發現設定它的權重為0也不影響收斂## The size, the iter number to run 訓練原始圖片大小、一次batch的樣本數、跑的epoch運行次數 image_size: 256 batch_size: 4 epoch: 2## Loss Network損失網絡 loss_model: "vgg_16" content_layers: # use these layers for content loss使用conv3_3定義內容損失- "vgg_16/conv3/conv3_3" style_layers: # use these layers for style loss使用conv1_2、conv2_2、conv3_3、conv4_3定義風格損失- "vgg_16/conv1/conv1_2"- "vgg_16/conv2/conv2_2"- "vgg_16/conv3/conv3_3"- "vgg_16/conv4/conv4_3" checkpoint_exclude_scopes: "vgg_16/fc" # we only use the convolution layers, so ignore fc layers.只用到卷積層所以不需要fc層 loss_model_file: "pretrained/vgg_16.ckpt" # the path to the checkpoint預訓練模型對應的位置

?

二、TensorBoard中監控訓練情況

?

?

?

?

總結

以上是生活随笔為你收集整理的CV之NS之VGG16：基于预训练模型VGG16训练COCO的train2014数据集实现训练《神奈川冲浪里》风格配置yml文件的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。