-2

我正在構建一個需要識別圖像中對象大小的應用程序。我需要能夠讀取黃瓜的圖片,然後告訴黃瓜有多長像素。我在過去建立了一個圖像分類器,但不知道如何檢索分類圖像並找到尺寸。如何檢測Sci-Kit Learn/Tensorflow中的對象大小?

我對機器學習和sci-kit學習有基本的瞭解。任何建議或指導將非常有幫助和讚賞!

編輯:這可能聽起來很愚蠢,但我可以扔任何東西到神經網絡來獲得自動結果嗎?例如,我可以訓練神經網絡的黃瓜圖像與他們的長度的目標,並期望神經網絡瞭解我想從圖像檢索像素長度?

+0

你指的*圖像分割*。圖像分割和圖像分類是機器學習「任務」的例子。 – guest

+0

感謝您的澄清!我已經更新了我的問題 – user3647894

+0

>例如,我可以訓練神經網絡的黃瓜圖像與他們的長度的目標,並期望神經網絡瞭解我想從圖像檢索像素長度? 是的,你可以。這被稱爲迴歸。最好的方法是首先建立一個分類DNN。之後剝離完全連接的圖層並將其替換爲用於迴歸的圖層,其中目標是像素數量。但是,您需要一大堆訓練數據 – CAFEBABE

回答

1

根據您的使用情況,使用OpenCV. It has bindings for python, C, C++ and java.可能會更好。如果你的黃瓜總是在同一背景上,這是我所要採用的100%的路線,因爲OpenCV擁有完全適合做這種事情的方法。

如果由於某種原因您必須在生產中使用TF,那麼您仍然應該使用OpenCV來準備數據以訓練神經網絡。準備一些黃瓜的示例圖像,使用OpenCV剪出黃瓜,並將其放入大約500,000個樣本圖像(縮放和旋轉隨機數量,也許有一點剪切或模糊或兩者),並將其用作數據集喂入TF。 您應該能夠輸入True/False值(黃瓜或不是黃瓜)以及大小。據我所知,它應該學習它們。

確保包含空白場景的大量圖像,然後包含與您的目標對象相同場景的圖像。理想情況下,您希望使用與您的應用程序相同的環境。

對不起,發佈答案而不是評論。還沒有足夠的聲望。

P.S.這是更容易檢測對象,如果你需要使用顏色而不僅僅是形狀。

乾杯, 安德魯

+0

感謝您的洞察!我將研究OpenCV。不幸的是,每次背景都會有所不同,我將不得不承認形狀而不僅僅是顏色。 – user3647894

+0

在OpenCV中的Haar級聯可能就足夠了。有教程位於這裏https://pythonprogramming.net/loading-images-python-opencv-tutorial/ –