๊ฒฐ๊ณผ๋ฅผ ์ธ์ํ ํ ์์ธก ๋ฐ ๊ฒฝ๊ณ ์์์ ํจ๊ป ์ด๋ฏธ์ง๋ฅผ ์ ์ฅํ๋ ๋ฐฉ๋ฒ์ด ์์ต๋๊น?
darkflow.net.build์์ TFNet ๊ฐ์ ธ์ค๊ธฐ
์์
cv2
์ต์ = {"๋ชจ๋ธ": "cfg/yolo.cfg", "๋ก๋": "bin/yolo.weights", "์๊ณ๊ฐ": 0.1}
tfnet = TFNet(์ต์ )
imgcv = cv2.imread("./sample_img/sample_dog.jpg")
๊ฒฐ๊ณผ = tfnet.return_predict(imgcv)
์ธ์(๊ฒฐ๊ณผ)
์์๋ฅผ ๊ทธ๋ฆฌ๊ณ json ์ถ๋ ฅ์ ๊ตฌ๋ฌธ ๋ถ์ํ ํ ์ด๋ฏธ์ง๋ฅผ ์์ฑํ๋ ํจ์๋ฅผ ์ง์ ์์ฑํด์ผ ํฉ๋๋ค.
๋ค์๊ณผ ๊ฐ์ ์ฝ๋๋ฅผ ์ฌ์ฉํ ์ ์์ต๋๋ค.
cv2.rectangle(imgcv,
(result["topleft"]["x"], result["topleft"]["y"]),
(result["bottomright"]["x"],
result["bottomright"]["y"]),
(0, 255, 0), 4)
text_x, text_y = result["topleft"][
"x"] - 10, result["topleft"]["y"] - 10
cv2.putText(imgcv, result["label"], (text_x, text_y),
cv2.FONT_HERSHEY_SIMPLEX, 0.8, (0, 255, 0), 2, cv2.LINE_AA)
์ฌ๊ธฐ์ result๋ return predict ๋ฉ์๋์ ์ํด ๋ฐํ๋ ๊ฒฐ๊ณผ ๋ชฉ๋ก์ ๋จ์ผ ์์์ด๊ณ imgcv๋ ์๋ณธ ์ด๋ฏธ์ง์ ๋๋ค. ์ด ๋์์ ๋ฐ๋๋๋ค!
๊ณ ๋ง์ต๋๋ค !
๊ฒฐ๊ณผ๋ฅผ ์ธ์ํ ํ ์์ธก ๋ฐ ๊ฒฝ๊ณ ์์์ ํจ๊ป ์ด๋ฏธ์ง๋ฅผ ์ ์ฅํ๋ ๋ฐฉ๋ฒ์ด ์์ต๋๊น?
darkflow.net.build์์ TFNet ๊ฐ์ ธ์ค๊ธฐ
์์ cv2์ต์ = {"๋ชจ๋ธ": "cfg/yolo.cfg", "๋ก๋": "bin/yolo.weights", "์๊ณ๊ฐ": 0.1}
tfnet = TFNet(์ต์ )
imgcv = cv2.imread("./sample_img/sample_dog.jpg")
๊ฒฐ๊ณผ = tfnet.return_predict(imgcv)
์ธ์(๊ฒฐ๊ณผ)
์ด ์ฝ๋๋ฅผ ์คํํ๋ ๋์ ๋ฌธ์ ๊ฐ ์์ต๋๋ค. "load" ๋งค๊ฐ๋ณ์์ ๋ํด ์๋ ค์ฃผ์ค ์ ์์ต๋๊น?
๊ฒฐ๊ณผ๋ฅผ ์ธ์ํ ํ ์์ธก ๋ฐ ๊ฒฝ๊ณ ์์์ ํจ๊ป ์ด๋ฏธ์ง๋ฅผ ์ ์ฅํ๋ ๋ฐฉ๋ฒ์ด ์์ต๋๊น?
darkflow.net.build์์ TFNet ๊ฐ์ ธ์ค๊ธฐ
์์ cv2
์ต์ = {"๋ชจ๋ธ": "cfg/yolo.cfg", "๋ก๋": "bin/yolo.weights", "์๊ณ๊ฐ": 0.1}
tfnet = TFNet(์ต์ )
imgcv = cv2.imread("./sample_img/sample_dog.jpg")
๊ฒฐ๊ณผ = tfnet.return_predict(imgcv)
์ธ์(๊ฒฐ๊ณผ)์ด ์ฝ๋๋ฅผ ์คํํ๋ ๋์ ๋ฌธ์ ๊ฐ ์์ต๋๋ค. "load" ๋งค๊ฐ๋ณ์์ ๋ํด ์๋ ค์ฃผ์ค ์ ์์ต๋๊น?
@abhishek795jha "load" ๋งค๊ฐ๋ณ์๋ ๊ฐ์ค์น ํ์ผ์ ๋๋ค........์ด๊ธฐ ์ฐ๋ฆฌ๋ ํ๋ จ ํ yolo ๊ฐ์ค์น ํ์ผ์ ์ฌ์ฉํ์ฌ ํ ์คํธ๋ฅผ ์ํด ๊ฐ์ค์น ํ์ผ๋ก .pb ํ์ผ์ ๋ง๋ญ๋๋ค. ํ๋ จ ์๊ฐ์ "--savepb" ๋ช ๋ น์ ์ฌ์ฉํ์ฌ with flow ๋ช ๋ น์ ์ฌ์ฉํ์ฌ .pb ํ์ผ์ ์ ์ฅํด์ผ ํฉ๋๋ค.
๊ฐ์ฅ ์ ์ฉํ ๋๊ธ
๋ค์๊ณผ ๊ฐ์ ์ฝ๋๋ฅผ ์ฌ์ฉํ ์ ์์ต๋๋ค.
์ฌ๊ธฐ์ result๋ return predict ๋ฉ์๋์ ์ํด ๋ฐํ๋ ๊ฒฐ๊ณผ ๋ชฉ๋ก์ ๋จ์ผ ์์์ด๊ณ imgcv๋ ์๋ณธ ์ด๋ฏธ์ง์ ๋๋ค. ์ด ๋์์ ๋ฐ๋๋๋ค!