Skip to content

๐ŸŽ  ๋จธ์‹ ๋Ÿฌ๋‹์„ ํ™œ์šฉํ•œ ์ ์ ํ†ต ์‹ค์‹œ๊ฐ„ ์ถ”์  ์‹œ์Šคํ…œ

Notifications You must be signed in to change notification settings

isieun0908/TFLite-Chamber-Object-Detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

2 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๋จธ์‹ ๋Ÿฌ๋‹์„ ํ™œ์šฉํ•œ ์ ์ ํ†ต ์‹ค์‹œ๊ฐ„ ์ถ”์  ์‹œ์Šคํ…œ

์š”์•ฝ

์ด ์—ฐ๊ตฌ์—์„œ๋Š” ๋ชจ๋ฐ”์ผ ์žฅ์น˜๋ฅผ ํ™œ์šฉํ•˜์—ฌ ๋‹ค์–‘ํ•œ ํ™˜๊ฒฝ์—์„œ ์ ์ ํ†ต์˜ ์ˆ˜์•ก ๋‚™ํ•˜๋ฅผ ํšจ๊ณผ์ ์œผ๋กœ ์ธ์‹ํ•˜๊ธฐ ์œ„ํ•ด์„œ ์ ์ ํ†ต ๊ฐ์ฒด ํƒ์ง€ ๋ชจ๋ธ์„ ์ƒ์„ฑํ•˜์˜€๋‹ค. ๋‹ค์–‘ํ•œ ๋ฐฐ๊ฒฝ์˜ ์ ์ ํ†ต ์‚ฌ์ง„์„ ์ˆ˜์ง‘ํ•˜๊ณ  ์ด๋ฅผ ๋ฐ”ํƒ•์œผ๋กœ ์ ์ ํ†ต์˜ ๋ฒ”์œ„ ์ง€์ •์„ ํ†ตํ•œ ๋ผ๋ฒจ๋ง์„ ํ•˜์—ฌ TF lite model maker์˜ ํ•™์Šต ๋ฐ์ดํ„ฐ๋กœ ์‚ฌ์šฉํ•˜์˜€๋‹ค. ์ƒ์„ฑํ•œ ๋ชจ๋ธ์„ TF Lite์— ์ ์šฉํ•˜์—ฌ ๋‹ค์–‘ํ•œ ํ™˜๊ฒฝ์—์„œ ์‹คํ—˜ํ•œ ๊ฒฐ๊ณผ ์† ๋–จ๋ฆผ๊ณผ ๋‹ค์–‘ํ•œ ์ฃผ๋ณ€ํ™˜๊ฒฝ ๊ฐ™์€ ๋…ธ์ด์ฆˆ์—์„œ ๋†’์€ ์ธ์‹์œจ์„ ๋ณด์—ฌ์ฃผ๋Š” ๊ฒƒ์„ ํ™•์ธํ•  ์ˆ˜ ์žˆ์—ˆ๋‹ค. ์ด ์‹คํ—˜์˜ ๊ฒฐ๊ณผ๋Š” ์ ์ ํ†ต ๋‚ด ์ˆ˜์•ก ๋‚™ํ•˜๋ฅผ ์œ„ํ•œ ๊ด€์‹ฌ์˜์—ญ ์„ค์ •์— ํ™œ์šฉ๋  ์ˆ˜ ์žˆ์„ ๊ฒƒ์œผ๋กœ ํŒ๋‹จ๋œ๋‹ค.

์—ฐ๊ตฌ ๊ณผ์ •

  1. ๋ฐ์ดํ„ฐ ์ˆ˜์ง‘
    ๊ฑฐ๋ฆฌ์™€ ๋ฐฐ๊ฒฝ์ด ๋‹ค๋ฅธ ํ™˜๊ฒฝ์—์„œ ์ดฌ์˜ํ•œ 79๊ฐœ์˜ ์˜์ƒ์—์„œ 1,307๊ฐœ์˜ ์‚ฌ์ง„ ๋ฐ์ดํ„ฐ๋ฅผ ์ถ”์ถœํ•˜์˜€๋‹ค.

  2. ๋ผ๋ฒจ๋ง
    LabelImg๋ฅผ ์ด์šฉํ•˜์—ฌ ์‚ฌ์ง„ ๋ฐ์ดํ„ฐ์—์„œ ์ ์ ํ†ต์˜ ๋ฒ”์œ„๋ฅผ ์ง€์ •ํ•˜์˜€์œผ๋ฉฐ, ๊ฐ์ฒด์˜ ์ด๋ฆ„์€ Chamber๋กœ ์„ค์ •ํ•˜์˜€๋‹ค. ๋ผ๋ฒจ๋ง์ด ์™„๋ฃŒ๋œ ๋ฐ์ดํ„ฐ๋Š” PascalVOCํ˜•์‹์˜ XMLํŒŒ์ผ๋กœ ๋ณ€ํ™˜ํ•˜์˜€๋‹ค.

  3. ํ•™์Šต๋ชจ๋ธ ์ƒ์„ฑ
    tensorflow lite model maker๋ฅผ ์ด์šฉํ•˜์—ฌ tflite๋ชจ๋ธ์„ ์ƒ์„ฑํ•˜์˜€๋‹ค. 1,307๊ฐœ์˜ ์‚ฌ์ง„๋ฐ์ดํ„ฐ ์ค‘ train์€ 1,110๊ฐœ, validate๋Š” 130๊ฐœ, test๋กœ 67๊ฐœ์˜ ๋ฐ์ดํ„ฐ๋ฅผ ์ด์šฉํ•˜์˜€๋‹ค.

image
[loss ๋ณ€ํ™”๋Ÿ‰]

  1. ๋ชจ๋ธ ์ ์šฉ ๋ฐ ์‹คํ–‰
    TFLite Object Detection์— ์ƒ์„ฑํ•œ Chamber.tflite๋ฅผ ์‚ฝ์ž…ํ•œ๋‹ค. ์ดํ›„ ๊น”๋”ํ•œ ๋ฐฐ๊ฒฝ 2๊ฐœ์™€ ๋ณต์žกํ•œ ๋ฐฐ๊ฒฝ 2๊ฐœ์˜ ๋ฐฐ๊ฒฝ์—์„œ ๊ฑฐ๋ฆฌ์™€ ์›€์ง์ž„์„ ๋ณ€ํ™”ํ•˜๋ฉฐ ์ธ์‹์˜ ์ •ํ™•๋„๋ฅผ ์‹คํ—˜ํ•˜์˜€๋‹ค.

  2. ๊ฒฐ๊ณผ
    ์ ๋‹นํ•œ ๊ฑฐ๋ฆฌ์—์„œ ํฐ ์›€์ง์ž„์—†์ด ์ธ์‹ํ•œ ๊ฒฝ์šฐ, ๋ฐฐ๊ฒฝ์— ์ƒ๊ด€์—†์ด ๋Œ€๋ถ€๋ถ„์˜ ์‹œ๊ฐ„๋™์•ˆ ์ตœ๊ณ  ์ •ํ™•๋ฅ ์„ ์œ ์ง€ํ•˜์—ฌ 99.61%์˜ ์ •ํ™•๋„๋ฅผ ๋ณด์˜€๋‹ค. ๊ฐ€๊นŒ์šด ๊ฑฐ๋ฆฌ์˜ ๊ฒฝ์šฐ ์ ์ ํ†ต์ด ๊ฐ€๊นŒ์ด์— ์žˆ์–ด ์ดˆ์ ์ด ๋งž์ง€ ์•Š์„ ๊ฒฝ์šฐ ์ •ํ™•๋ฅ ์ด ๋–จ์–ด์ ธ์„œ ์ „์ฒด์ ์ธ ์ •ํ™•๋ฅ ๋„ ๋‚ฎ์•„์ง„ ๊ฒƒ์„ ํ™•์ธํ•  ์ˆ˜ ์žˆ๋‹ค. ๋จผ ๊ฑฐ๋ฆฌ์—์„œ๋Š”, ์ ์ ํ†ต์ด ์ž‘๊ฒŒ ๋ณด์—ฌ ์ธ์‹ ์ •ํ™•๋„๋Š” ์กฐ๊ธˆ ๋–จ์–ด์กŒ๋‹ค. ์›€์ง์ด๋ฉด์„œ ์ธ์‹ํ•  ๋•Œ๋Š” ๋ณต์žกํ•œ ๋ฐฐ๊ฒฝ์ผ ๋•Œ๋Š” ์ •ํ™•๋ฅ ์ด ๋–จ์–ด์ง€๋Š” ๊ฒฝํ–ฅ์„ ๋ณด์˜€์œผ๋‚˜ 96%์ด์ƒ์˜ ์ •ํ™•๋ฅ ์„ ๋ณด์ด๋ฉด์„œ ์›€์ง์ž„์ด ์žˆ์„ ๋•Œ์—๋„ ์‹ค์‹œ๊ฐ„์œผ๋กœ ์ •ํ™•ํžˆ ์ถ”์ ์ด ๊ฐ€๋Šฅํ•œ ๊ฒƒ์„ ํ™•์ธํ•˜์˜€๋‹ค.

About

๐ŸŽ  ๋จธ์‹ ๋Ÿฌ๋‹์„ ํ™œ์šฉํ•œ ์ ์ ํ†ต ์‹ค์‹œ๊ฐ„ ์ถ”์  ์‹œ์Šคํ…œ

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages