亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

帶有 tf.image.rgb_to_grayscale 的自定義預處理函數

帶有 tf.image.rgb_to_grayscale 的自定義預處理函數

海綿寶寶撒 2021-08-24 18:19:22
我正在嘗試使用自定義預處理函數在訓練期間將 RGB 圖像轉換為灰度。因此,我嘗試使用tf.image.rbg_to_grayscale它。我的功能如下所示:def prep_data(x):    x = tf.image.rgb_to_grayscale(x)    return xdatagen = ImageDataGenerator(preprocessing_function=prep_data,validation_split=0.15)的train_generator是使用限定datagen.flow_from_dataframe(...)。沒有這個自定義函數的訓練效果很好,但是一旦我使用它,我就會收到以下錯誤:ValueError:使用序列設置數組元素。從這個情況來看答案在這里,我想我需要我的輸入改變rgb_to_grayscale,但我不知道什么是合格的正確方法x的功能。關于如何解決這個問題的任何想法?
查看完整描述

2 回答

?
婷婷同學_

TA貢獻1844條經驗 獲得超8個贊

相反,您可以使用color_mode參數 offlow_from_directory并將其設置'grayscale'為將圖像轉換為灰度。來自Keras 文檔

color_mode:“灰度”、“rbg”、“rgba”之一。默認值:“RGB”。圖像是否將轉換為具有 1、3 或 4 個通道。


查看完整回答
反對 回復 2021-08-24
?
倚天杖

TA貢獻1828條經驗 獲得超3個贊

將 RGB 圖像更改為灰度(示例):

img = image.load_img('/kaggle/input/cassava-leaf-disease-classification/train_images/'+train['image_id'][i], target_size=(28,28,1), color_mode="grayscale")



查看完整回答
反對 回復 2021-08-24
  • 2 回答
  • 0 關注
  • 408 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號