predict_show_price.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[],"authorship_tag":"ABX9TyOQ/g7Spvwwjh8rrsoS+PsC"},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"}},"cells":[{"cell_type":"code","source":["!pip install ultralytics\n","cls_names = {0: 'binch', 1: 'oreoblack', 2: 'oreored', 3: 'cushcush', 4: 'gosomi', 5: 'yegamcheese', 6: 'oreothinstaramisu', 7: 'oreothinsvanillamooth', 8: 'chocchochanchocochip', 9: 'chicchoc', 10: 'yuginongblueberrytt', 11: 'yuginongaroniatt', 12: 'yuginongsiguemchitt', 13: 'yuginongbananatt', 14: 'yuginongplaneyogurtcube', 15: 'yuginongbluberryyogurtcube', 16: 'yuginongbananayogurtcube', 17: 'yuginongddalgiyogurtcube', 18: 'finggercheeseblueberry', 19: 'finggercheeseorigin', 20: 'ggobuk', 21: 'simpop', 22: 'cheetosr', 23: 'cheetosb', 24: 'cheetosy', 25: 'gamjasokb', 26: 'gamjatokb', 27: 'gamjatokg', 28: 'beksulgangryeokmilgaru', 29: 'beksulbakryeokmilgaru', 30: 'beksuljungryeokmilgaru', 31: 'beksulchalmilgaru', 32: 'beksulwooriddangwoorimealmilgaru', 33: 'beksulbuchimgaru', 34: 'beksultwigimgaru', 35: 'beksulbasakbuchimgaru', 36: 'beksulbasaktwigimgaru', 37: 'beksulchickentwigimgaru', 38: 'gompyomilgarujoongryukdamokjukyong', 39: 'gompyochaljinhaetmilgaru', 40: 'gompyobuchimgaru', 41: 'ottogitwigimgaru', 42: 'ottogippanggaru', 43: 'nonghyupguksan100%ssalbuchimgaru', 44: 'beksulmc&cheesekit', 45: 'beksulpancakemix', 46: 'beksulhotcakemix', 47: 'beksulyeongyanggyunhyeonghotcakemix', 48: 'beksulwoorimealhotcakemix', 49: 'beksulchocochiphotcakemix', 50: 'beksulchococakemix', 51: 'beksulsikppangmix', 52: 'beksuldonutmix', 53: 'beksulmuffinmix', 54: 'beksulbrowniemix', 55: 'beksulchapssalhotteokmix', 56: 'beksulnokchamatchapssalhotteokmix', 57: 'beksulinjeolmihotteokmix', 58: 'beksulchocochipcookiemix', 59: 'ottogichapssalhotteokmix', 60: 'ottogihotcakemix', 61: 'ottogissalhotcakemix', 62: 'ottogichoganpyeonkimchijeonmix', 63: 'damoolchamimatbakingpowder', 64: 'damoolchamimatnewsugar', 65: 'damoolsoda', 66: 'damoolgyepigaru', 67: 'qonehomemadeyungyangcookiemix', 68: 'qonehomemademuffinmix', 69: 'qonehomemadebrowniemix', 70: 'a1steaksauce', 71: 'ottogigosohanmayonnaise', 72: 'ottogifreshmayonnaise', 73: 'chungjungonechamkkaeheukimja', 74: 'ottogiwasabidressing', 75: 'ottogiorientaldressing', 76: 'fontanaoriental', 77: 'fontanazerofatgreenkiwi', 78: 'fontanazerofatpineapple', 79: 'ottogicaloriereuljurinketchup33%', 80: 'ottogijalapenoketchup', 81: 'srirachaketchup', 82: 'ottogigosohangoldmayonnaise', 83: 'chungjeongwonsweetcornmayosauce', 84: 'chungjeongwongarliccheesemayosauce', 85: 'ottogicoleslawdressing', 86: 'chungjeongwoncoleslawdressing', 87: 'homeplusthousandIslanddressing', 88: 'homepluskiwidressing', 89: 'homepluschamkkaedressing', 90: 'homeplushoneymustard', 91: 'homeplusbalsamicdressing', 92: 'sampioyeondoosoon', 93: 'chungjungonesoonchanggochujang', 94: 'chungjungonesoonchangchogochujang', 95: 'chungjungonesoonchangtaeyangchossalgochujang', 96: 'chungjungonesoonchangsogogibibimgochujang', 97: 'chungjungonesoonchangsogogibokkeumgochujang', 98: 'chungjungonesoonchangjaeraesikdoenjang', 99: 'chungjungonesoonchangmisodoenjang', 100: 'chungjungonesoonchangyangnyeomdeumppukssamjang', 101: 'chungjungonesoonchanggogijeonyongssamjang', 102: 'chungjungonesoonchanggogijeonyonggochujang', 103: 'chungjungonesoonchang100%hyeonmitaeyangchochalgochujang', 104: 'chungjungonesoonchang100%hyeonmitaeyangchomaewoongochujang', 105: 'chungjungonesoonchangmakjang', 106: 'chungjungonemasinneunjoonghwachunjang', 107: 'haechandlewoorissalromandeuntaeyangchogoldgochujang', 108: 'haechandlewoorissalromandeunmaewoontaeyangchogoldgochujang', 109: 'haechandlechogochujang', 110: 'haechandlejaeraesikdoenjang', 111: 'haechandlesagyejeolssamjang', 112: 'haechandle8gajiguksanyangnyeomssamjang', 113: 'haechandlegeudaerodoenjangjjigaesogogi', 114: 'haechandlesogogibokkeumgochujang', 115: 'sempiobaekildoenjang', 116: 'sempiotojang', 117: 'jinmichunjang', 118: 'hongcho100%cheonyeonbalhyoseokryu', 119: 'hongcho100%cheonyeonbalhyoblueberry', 120: 'beksuljayeondameunbalhyoapplesikcho', 121: 'beksuljayeondameunbalhyolemonsikcho', 122: 'beksulfructooligodang', 123: 'beksulssaloligodang', 124: 'beksulyorioligodang', 125: 'chungjungonemullyeot', 126: 'chungjungonejocheongssalyeot', 127: 'sempiojinganjangguemf3', 128: 'sempioyangjoganjang501', 129: 'sempioyangjoganjang701', 130: 'sempiohatsaldameunjorimganjang', 131: 'sempiohatsaldameungukganjang', 132: 'sempiomatganjang', 133: 'sempiomasitgeyeomdonajeunyangjoganjang', 134: 'chungjungonejejudomoseulpomyeolchi', 135: 'beksulhayansultang', 136: 'beksulgalsaeksultang', 137: 'beksulheuksultang', 138: 'beksulyukinonggalsaeksultang', 139: 'beksulmomeheupsureuljurinxylosesultang', 140: 'beksulmomeheupsureuljurinbrownxylosesultang', 141: 'chungjungoneyukinonghwangsultang', 142: 'beksuldwaejigalbiyangnyeom', 143: 'chungjungonehaemulgulsauce', 144: 'beksulnamhaegulsauce', 145: 'beksulrosespaghettisauce', 146: 'hasunjungmyeolchiaekjeot', 147: 'hasunjungseohaeankkanariaekjeot', 148: 'chungjungonedaenamusukseongmyeolchiaekjeot', 149: 'chungjungoneseohaeankkanarigold', 150: 'heukwondangheukdangsyrup', 151: 'chungjungonealfredosaengcream&cheese', 152: 'chungjungonealfredobacon&cream', 153: 'chungjungonealfredomyeongran&saengcream', 154: 'chungjungonealfredogarlic&peperoncino', 155: 'chungjungonealfredojogae&whitewine', 156: 'beksulgalbiyangnyeom', 157: 'beksulsariwonbulgogi', 158: 'chungjungonedambaekanbulgogiyangnyeom', 159: 'chungjungonesogalbiyangnyeom', 160: 'dececcobazilpesto', 161: 'heinzclassicobazilpesto', 162: 'bernibazilpesto', 163: 'bernipomodoropesto', 164: 'fontanagenovacheese&cream', 165: 'fontanacarniabacon&mushroomcream', 166: 'fontanaparmaquattrocheesecream', 167: 'baeksulnamhaegulsauce', 168: 'baeksuljeonbokgulsauce', 169: 'baeksulgulsaucehaemulmat', 170: 'leekumkeepandagulsauce', 171: 'leekumkeepremiumgulsauce', 172: 'leekumkeemapasauce', 173: 'leekumkeechickenpowder', 174: 'leekumkeenongchukchickensauce', 175: 'leekumkeehuoguomaratangsauce', 176: 'paikcookmanneungbokkeumyorisauce', 177: 'chungjeongwonpremium gulsauce', 178: 'noganmapungmiyuralcho gochugireum', 179: 'ottogirajojangyangpa', 180: 'ottogirajojangsancho', 181: 'chungjeongwonyujaponssauce', 182: 'moranbongshabushabusauce', 183: 'kikkomanyujapons', 184: 'himornchillishabusukki', 185: 'haidilaomarasyanggwosauce', 186: 'haidilaomarahuoguosauce', 187: 'huoguodiryoshabuyuksu', 188: 'sampioguksijanggukkatsuo', 189: 'sampioguksijanggukmyeolchidasima', 190: 'baeksulsaengsunjorimyangnyeom', 191: 'ottogifrescotomatospaghettisauce', 192: 'ottogifrescomeatspaghettisauce', 193: 'ottogifrescorosespaghettisauce', 194: 'ottogifrescomaneul&yangpaspaghettisauce', 195: 'ottogifrescotrufflecreamspaghettisauce', 196: 'cjsandeuraecheoeumbuteoyuksuyuksuhwangtae', 197: 'cjsandeuraecheoeumbuteoyuksuyuksukatsuo', 198: 'cjsandeuraecheoeumbuteoyuksuyuksumyeolchidasima', 199: 'ottogibbqsauce', 200: 'ottogisteaksauce', 201: 'ottogiworcestersauce', 202: 'heinzclassicogarlicalfredo', 203: 'heinzclassicoitaliansausage', 204: 'cjdasidasogogi', 205: 'cjdasidamyeolchi', 206: 'cjsandeuraehanwoodasida', 207: 'simplusyasaenghwaggul', 208: 'miwongamchilmatmiwon', 209: 'miwon matsogeum', 210: 'beksulcanolayu', 211: 'beksulpodoseeyu', 212: 'beksuloliveoil', 213: 'beksulgungangyoriyu', 214: 'heapyocanolayu', 215: 'heapyokkwakchanoliveyu', 216: 'heapyopodoseedyu', 217: 'chungjungoneoksusuyu', 218: 'chungjungonecanolayu', 219: 'ottogifrescooliveyu', 220: 'ottogiyetnalchamgireum', 221: 'beksuljinhanchamgireum', 222: 'ottogiyetnalchamgireum', 223: 'ottogigosohanchamgireum', 224: 'baeksulgosohamgadeukchamgireum', 225: 'haepyokonggireum', 226: 'chungjeongwonsinancheonilyeommatsogeum', 227: 'chungjeongwonsinancheonilyeomguwoonsogeum', 228: 'chungjeongwonsinancheonilyeomganeunsogeum', 229: 'baeksulcheonilyummatsogeum', 230: 'baeksulmatsogeum', 231: 'baeksulggotsogeum', 232: 'baeksulguwoonsogeum', 233: 'haepyoggotsogeum', 234: 'daesangsogogigamchimi', 235: 'baeksulxylosesultang', 236: 'baeksulxylosebrownsultang', 237: 'baeksulyukinonggalsaeksultang', 238: 'nokchawonsayangbulggul', 239: 'ggotsaemayangbulggul', 240: 'sempiossamtojang', 241: 'chungjungonesoonchangmyunggassamjang', 242: '365cafeelpasovanillasyrup', 243: 'beksulyoridang', 244: 'leggosbasilpesto', 245: 'leggoschilli&garlicpesto', 246: 'ottogimasinneunbookuhguk', 247: 'ottogiganpyeonmiyeokguk', 248: 'ottogiganpyeonbookeotguk', 249: 'ottogiganpyeonsagolugeojiguk', 250: 'ottogiyetnaldakgomtang', 251: 'ottogiyetnalyukgaejang', 252: 'ottogiyetnalsagolgomtang', 253: 'ottogiyetnalgalbitang', 254: 'ottogiyetnaldoganitang', 255: 'ottogiyetnalteukyukgaejang', 256: 'ottogiandongsiksogogigukbab', 257: 'ottogiseooulsikseolleongtang', 258: 'ottoginamdosikhanwoomiyeokguk', 259: 'ottogibusansikdwaejigukbabgomtang', 260: 'ottogijongrosikdoganitang', 261: 'ottogiyetnalgusuhannurungji', 262: 'yangbanjinguksagolgomtang', 263: 'yangbanjinguksogogimuguk', 264: 'yangbanjinguksagolgomtang', 265: 'yangbanhanwoosagolseolleongtang', 266: 'yangbanchadolyukgaejang', 267: 'yangbanchadoldoenjangjjigae', 268: 'yangbanchamchibudaejjigae', 269: 'yangbantongchamchikimchijjigae', 270: 'yangbandwaejigogikimchijjigae', 271: 'yangbanjeonjusikhwangtaecongnamulguk', 272: 'yangbanjingukdakgomtang', 273: 'yangbanbaekapmiyeokguk', 274: 'bibigosamgyetang', 275: 'bibigoyukgaejang', 276: 'bibigochadolyukgaejang', 277: 'bibigosagolgomtang', 278: 'bibigosogogimiyeokguk', 279: 'bibigodakgomtang', 280: 'bibigosoondaetguk', 281: 'bibigochueotang', 282: 'bibigospambudaejjigae', 283: 'bibigoseolleongtang', 284: 'bibigoaltang', 285: 'bibigodwaejigogikimchijjigae', 286: 'jonggajonggabansanggopchangyukgaejang', 287: 'jonggajonggabansangbeoseotdeulkkaemiyeokguk', 288: 'jonggajonggabansangeolkeunaltang', 289: 'chungjungoneilsanggajeongsikhanwoojinyuksu', 290: 'jonggajonggabansanggopchangyukgaejang', 291: 'jonggajonggabansangsagolsunjihaejangguk', 292: 'jonggajonggabansangjeontongsoondaetguk', 293: 'chungjungoneilsanggajeongsiksagolkimchijjigae', 294: 'chungjungoneilsanggajeongsikeolkeunsoondubujjigae', 295: 'chungjungoneilsanggajeongsiksagolsiraegidoenjangguk', 296: 'chungjungoneilsanggajeongsiksagolugeojideulkkaetang', 297: 'chungjungoneilsanggajeongsikbeoseotdeulkkaemiyeokguk', 298: 'chungjungoneilsanggajeongsikhanwoojingomtang', 299: 'cookteamdwaejigalbijeongol', 300: 'KSBoneureunsogipyeonhannurungji', 301: 'bibigosogogijangteoguk', 302: 'bibigosogogimuguk', 303: 'ottogidwaejigogikimchijjigae', 304: 'bibigogalbitang', 305: 'otttogiuijeongbubudaejjigae', 306: 'ottogisuwonsikugeojigalbitang', 307: 'bibigohanwoosagolgomtang', 308: 'homeplusdwaejigogijangjorim', 309: 'homeplusbutterjangjorim', 310: 'homeplusjinmichaealmondjorim', 311: 'homeplusyeongeunjorim', 312: 'homeplusgyeongwaryumyeolchibokkeum', 313: 'homepluskongjaban', 314: 'homeplusyangnyeomkkaetnip', 315: 'homeplusmaneuljjongmuchim', 316: 'homeplusmumallaengimuchim', 317: 'homeplusmaechurialjangjorim', 318: 'bonjuksogogijangjorim', 319: 'bonjuknamdokkomakmuchim', 320: 'bonjukmaechurialjangjorim', 321: 'bonjukjejuansimjangjorim', 322: 'jonggadwaejigogimaechurialjangjorim', 323: 'jonggasogogimaechurialjangjorim', 324: 'jonggagyeongwaryumyeolchibokkeum', 325: 'jonggagochujangmyeolchibokkeum', 326: 'jonggaojingeochaebokkeum', 327: 'jonggageomeunkongjorim', 328: 'jonggayeongeunjorim', 329: 'bibigogodeungeosiraegijorim', 330: 'bibigokkongchikimchijorim', 331: 'bibigokodarimujorim', 332: 'ottogimaekomkkongchijorim', 333: 'ottogipomodorokkongchijorim', 334: 'ottogikimchigodeungeojorim', 335: 'lottehamgochujanggogibokkeum', 336: 'lottehamjangjorim', 337: 'yorihadamiyeokguk', 338: 'yorihadahwangtaehaejangguk', 339: 'bibigogamjatang', 340: 'bibigochadoldoenjangjjigae', 341: 'bibigojeonbokmiyeokguk', 342: 'bibigojingukseolleongtang', 343: 'bibigodubukimchijjigae', 344: 'bibigodoenjangjjigae', 345: 'bibigobangyetang', 346: 'bibigokongnamulhwangtaeguk', 347: 'yorihadaeolkeunsoondaeguk', 348: 'yorihadakimchijjigae', 349: 'yorihadayukgaejang', 350: 'yorihadasagolgomtang', 351: 'yorihadayangjiyuksu', 352: 'yorihadasumgyetang', 353: 'yorihadasusambangyetang', 354: 'yangbanboyangsamgyetang', 355: 'yangbanhanwoosagolsiraegiguk', 356: 'yangbanaehobakgochujangjjigae', 357: 'yangbanjeonjusikhwangtaekongnamulguk', 358: 'yangbanugeojidoenjangguk', 359: 'dongwonganpyeonyorikitgolbaengibibimureonggangdoenjang', 360: 'dongwonganpyeonyorikitgolbaengimulhoe', 361: 'dongwonganpyeonyorikitgolbaengitangeolkeunhangungmul', 362: 'dongwonganpyeonyorikitureonggangdoenjang', 363: 'nongshimneogurimild', 364: 'nongshimneogurispicy', 365: 'nongshimneoguriangry', 366: 'nongshimgamjamyeon', 367: 'nongshimjjawang', 368: 'nongshimjjawangspicy', 369: 'nongshimdoongjinaengmyeonbibim', 370: 'nongshimdoongjinaengmyeondongchimi', 371: 'ottogijinramenmild', 372: 'ottogijinramenspicy', 373: 'yangbanchamchijuk', 374: 'yangbanyachaejuk', 375: 'yangbansogokijuk', 376: 'yangbandeulkkaedakjuk', 377: 'yangbanbaekhabjuk', 378: 'yangbandanhobakjuk', 379: 'yangbanbamdanpatjuk', 380: 'yangbanmiyeokjuk', 381: 'yangbanjeonbokjuk', 382: 'yangbanmpjeonbokjuk', 383: 'yangbanmpbaekhabjuk', 384: 'yangbansamgyejeonbokjuk'}"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"-cNKaG1xh2cu","executionInfo":{"status":"ok","timestamp":1693975986461,"user_tz":-540,"elapsed":5522,"user":{"displayName":"김건형","userId":"08017677761814619335"}},"outputId":"c5a6ff27-cfd5-4e8c-bb71-556a0a5c7e35"},"execution_count":138,"outputs":[{"output_type":"stream","name":"stdout","text":["Requirement already satisfied: ultralytics in /usr/local/lib/python3.10/dist-packages (8.0.171)\n","Requirement already satisfied: matplotlib>=3.2.2 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (3.7.1)\n","Requirement already satisfied: numpy>=1.22.2 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (1.23.5)\n","Requirement already satisfied: opencv-python>=4.6.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (4.8.0.76)\n","Requirement already satisfied: pillow>=7.1.2 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (9.4.0)\n","Requirement already satisfied: pyyaml>=5.3.1 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (6.0.1)\n","Requirement already satisfied: requests>=2.23.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (2.31.0)\n","Requirement already satisfied: scipy>=1.4.1 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (1.10.1)\n","Requirement already satisfied: torch>=1.8.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (2.0.1+cu118)\n","Requirement already satisfied: torchvision>=0.9.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (0.15.2+cu118)\n","Requirement already satisfied: tqdm>=4.64.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (4.66.1)\n","Requirement already satisfied: pandas>=1.1.4 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (1.5.3)\n","Requirement already satisfied: seaborn>=0.11.0 in /usr/local/lib/python3.10/dist-packages (from ultralytics) (0.12.2)\n","Requirement already satisfied: psutil in /usr/local/lib/python3.10/dist-packages (from ultralytics) (5.9.5)\n","Requirement already satisfied: py-cpuinfo in /usr/local/lib/python3.10/dist-packages (from ultralytics) (9.0.0)\n","Requirement already satisfied: contourpy>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (1.1.0)\n","Requirement already satisfied: cycler>=0.10 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (0.11.0)\n","Requirement already satisfied: fonttools>=4.22.0 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (4.42.1)\n","Requirement already satisfied: kiwisolver>=1.0.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (1.4.4)\n","Requirement already satisfied: packaging>=20.0 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (23.1)\n","Requirement already satisfied: pyparsing>=2.3.1 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (3.1.1)\n","Requirement already satisfied: python-dateutil>=2.7 in /usr/local/lib/python3.10/dist-packages (from matplotlib>=3.2.2->ultralytics) (2.8.2)\n","Requirement already satisfied: pytz>=2020.1 in /usr/local/lib/python3.10/dist-packages (from pandas>=1.1.4->ultralytics) (2023.3)\n","Requirement already satisfied: charset-normalizer<4,>=2 in /usr/local/lib/python3.10/dist-packages (from requests>=2.23.0->ultralytics) (3.2.0)\n","Requirement already satisfied: idna<4,>=2.5 in /usr/local/lib/python3.10/dist-packages (from requests>=2.23.0->ultralytics) (3.4)\n","Requirement already satisfied: urllib3<3,>=1.21.1 in /usr/local/lib/python3.10/dist-packages (from requests>=2.23.0->ultralytics) (2.0.4)\n","Requirement already satisfied: certifi>=2017.4.17 in /usr/local/lib/python3.10/dist-packages (from requests>=2.23.0->ultralytics) (2023.7.22)\n","Requirement already satisfied: filelock in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (3.12.2)\n","Requirement already satisfied: typing-extensions in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (4.7.1)\n","Requirement already satisfied: sympy in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (1.12)\n","Requirement already satisfied: networkx in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (3.1)\n","Requirement already satisfied: jinja2 in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (3.1.2)\n","Requirement already satisfied: triton==2.0.0 in /usr/local/lib/python3.10/dist-packages (from torch>=1.8.0->ultralytics) (2.0.0)\n","Requirement already satisfied: cmake in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.8.0->ultralytics) (3.27.2)\n","Requirement already satisfied: lit in /usr/local/lib/python3.10/dist-packages (from triton==2.0.0->torch>=1.8.0->ultralytics) (16.0.6)\n","Requirement already satisfied: six>=1.5 in /usr/local/lib/python3.10/dist-packages (from python-dateutil>=2.7->matplotlib>=3.2.2->ultralytics) (1.16.0)\n","Requirement already satisfied: MarkupSafe>=2.0 in /usr/local/lib/python3.10/dist-packages (from jinja2->torch>=1.8.0->ultralytics) (2.1.3)\n","Requirement already satisfied: mpmath>=0.19 in /usr/local/lib/python3.10/dist-packages (from sympy->torch>=1.8.0->ultralytics) (1.3.0)\n"]}]},{"cell_type":"code","execution_count":139,"metadata":{"id":"d8u1gs1caUCI","executionInfo":{"status":"ok","timestamp":1693975986462,"user_tz":-540,"elapsed":7,"user":{"displayName":"김건형","userId":"08017677761814619335"}}},"outputs":[],"source":["import cv2\n","from ultralytics import YOLO\n","from google.colab.patches import cv2_imshow\n","import cv2"]},{"cell_type":"code","source":["model = YOLO('best.pt')\n","img = 'custom_image'\n","results = model.predict(img + '.jpg', save_txt=True)"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"1PCYTOGkaeuw","executionInfo":{"status":"ok","timestamp":1693976152652,"user_tz":-540,"elapsed":2730,"user":{"displayName":"김건형","userId":"08017677761814619335"}},"outputId":"a9d26e68-29ba-460b-f064-acc6cbe03019"},"execution_count":144,"outputs":[{"output_type":"stream","name":"stderr","text":["\n","image 1/1 /content/05-450.jpg: 352x608 3 simpops, 3 cheetosrs, 5 cheetosbs, 1 cheetosy, 416.9ms\n","Speed: 3.6ms preprocess, 416.9ms inference, 3.2ms postprocess per image at shape (1, 3, 352, 608)\n","Results saved to \u001b[1mruns/detect/predict11\u001b[0m\n","1 label saved to runs/detect/predict11/labels\n"]}]},{"cell_type":"code","source":["products_prices = {'cheetosb': 1300, 'cheetosr': 1300, 'cheetosy': 1300, 'simpop': 1200, 'gamjatokb': 1400, 'gamjatokg': 1400, 'ggobuk': 1100, 'ottogijinramenmild': 800}\n","dir = str(results[0]).split('\\n')[-2].split(\"'\")[1]\n","with open(dir + '/labels/{}.txt'.format(img)) as f:\n","    lines = f.readlines()\n","f.close()\n","count_result = {}\n","for line in lines:\n","    obj = int(line.split(' ')[0])\n","    if obj in count_result:\n","        count_result[obj] += 1\n","    else:\n","        count_result[obj] = 1"],"metadata":{"id":"Q9JvdHJhi1t8","executionInfo":{"status":"ok","timestamp":1693976154212,"user_tz":-540,"elapsed":417,"user":{"displayName":"김건형","userId":"08017677761814619335"}}},"execution_count":145,"outputs":[]},{"cell_type":"code","source":["res_plotted = results[0].plot()\n","cv2_imshow(res_plotted)\n","\n","total = 0\n","for product_code, count in count_result.items():\n","    name = cls_names[product_code]\n","    print('{}: {}개. (1개당 {}원)'.format(name, count, products_prices[name]))\n","    total += products_prices[name] * count\n","print('총 금액: {}원'.format(total))"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":591,"output_embedded_package_id":"1ELMZKqhoJqqClAqqMbRbkp4D1GUVFElg"},"id":"tohhs1wWra_H","executionInfo":{"status":"ok","timestamp":1693976156540,"user_tz":-540,"elapsed":2036,"user":{"displayName":"김건형","userId":"08017677761814619335"}},"outputId":"b130ce46-0031-4c6c-d4b3-142b21429a31"},"execution_count":146,"outputs":[{"output_type":"display_data","data":{"text/plain":"Output hidden; open in https://colab.research.google.com to view."},"metadata":{}}]},{"cell_type":"code","source":[],"metadata":{"id":"GvwVcLHTwzXx","executionInfo":{"status":"ok","timestamp":1693976011438,"user_tz":-540,"elapsed":1,"user":{"displayName":"김건형","userId":"08017677761814619335"}}},"execution_count":142,"outputs":[]}]}