/farmer

farmer is an automated machine learning library.๐Ÿ‘จโ€๐ŸŒพ

Primary LanguagePythonApache License 2.0Apache-2.0

farmer

You can train Classification and Segmentation tasks as best practice

Requirements: ncc

$ git clone git@github.com:NCC-AI/ncc.git
$ cd ncc
$ python setup.py install

Installation

$ pip install git+https://github.com/NCC-AI/farmer

Prepare Data set folder

  • classification

    โ”‚ โ”œโ”€โ”€ target_directory โ”‚ โ”œโ”€โ”€ data_case_directory(dataA) # caseใ”ใจใ ใฃใŸใ‚Šใ€train/testใ ใฃใŸใ‚Š โ”‚ โ”‚ โ”‚ โ”œโ”€โ”€ category_directory(Orange) # ใ‚ฏใƒฉใ‚นใฎใƒ•ใ‚ฉใƒซใƒ€ใƒผ โ”‚ โ”‚ โ”‚
    โ”‚ โ”‚ โ”œโ”€โ”€ image_file(jpg or png) โ”‚ โ”‚ โ”‚ โ”œโ”€โ”€ category_directory(Apple) โ”‚ โ”‚
    โ”‚ โ”œโ”€โ”€ image_file(jpg or png) โ”‚
    โ”œโ”€โ”€ data_case_directory(dataB)

  • segmentation

    โ”‚ โ”œโ”€โ”€ target_directory โ”‚ โ”œโ”€โ”€ data_case_directory(dataA) # caseใ”ใจใ ใฃใŸใ‚Šใ€train/testใ ใฃใŸใ‚Š โ”‚ โ”‚ โ”‚ โ”œโ”€โ”€ input_image_directory # ๅ…ฅๅŠ›็”ปๅƒใƒ•ใ‚ฉใƒซใƒ€ โ”‚ โ”‚ โ”‚
    โ”‚ โ”‚ โ”œโ”€โ”€ image_file(jpg or png) โ”‚ โ”‚ โ”‚ โ”œโ”€โ”€ mask_image_directory # ใƒžใ‚นใ‚ฏ็”ปๅƒใƒ•ใ‚ฉใƒซใƒ€ โ”‚ โ”‚
    โ”‚ โ”œโ”€โ”€ image_file(jpg or png) โ”‚
    โ”œโ”€โ”€ data_case_directory(dataB)

Training

config.iniใซๅญฆ็ฟ’ๆกไปถใ‚’ๆ›ธใ่พผใฟใ€ใ“ใฎใƒ•ใ‚กใ‚คใƒซใŒใ‚ใ‚‹ๅ ดๆ‰€ใงไปฅไธ‹ใฎใ‚ณใƒžใƒณใƒ‰ใ‚’ๅฎŸ่กŒ

$ ncc-cls  # classification
$ ncc-seg  # segmentation

secret.iniใซใ‚นใƒฉใƒƒใ‚ฏใฎๆƒ…ๅ ฑใ‚’ๆ›ธใ่พผใ‚ใฐใ€ใƒญใ‚ฐ็”ปๅƒใ‚’้ฃ›ใฐใ›ใ‚‹ใ€‚

Result

  • ๅฎŸ่กŒใ—ใŸๆ—ฅไป˜ๆ™‚้–“ใง่‡ชๅ‹•ใซใƒ•ใ‚ฉใƒซใƒ€ใ‚’ไฝœๆˆใ€‚็ตๆžœใ‚’ไปฅไธ‹ใฎใƒ‡ใ‚ฃใƒฌใ‚ฏใƒˆใƒชๆง‹้€ ใงไฟๅญ˜ใ•ใ‚Œใพใ™ใ€‚

    โ”‚ โ”œโ”€โ”€ result โ”œโ”€โ”€ ๆ—ฅไป˜ๆ™‚้–“(็ตๆžœA) โ”‚ โ”œโ”€โ”€ image # ๆŽจ่ซ–ใ‚ตใƒณใƒ—ใƒซ โ”‚ โ”œโ”€โ”€ info # ่จญๅฎšใƒ•ใ‚กใ‚คใƒซ/็”ปๅƒใƒ‘ใ‚น โ”‚ โ”œโ”€โ”€ learning # ๅญฆ็ฟ’ๅฑฅๆญด โ”‚ โ”œโ”€โ”€ model # ๆœ€่‰ฏใƒขใƒ‡ใƒซ & ๆœ€็ต‚ใƒขใƒ‡ใƒซ โ”‚
    โ”œโ”€โ”€ ๆ—ฅไป˜ๆ™‚้–“(็ตๆžœB)