このページの本文へ

ディープラーニング フレームワーク「SmallTrain ver. 0.2.0」の ソースコードをGitHubに公開

株式会社 Geek Guild
2020年11月10日

  • この記事をはてなブックマークに追加
  • 本文印刷

株式会社 Geek Guild
【商用利用可AIオープンソース】高精度なAI学習済みモデルの開発に使える AIモデルとライブラリを完全公開しました!

株式会社 Geek Guild(ギークギルド)(本社:京都市)は、商用AIサービスのための、高精度なAI学習済みモデルの開発に使えるディープラーニングフレームワーク「SmallTrain(スモールトレイン)ver.0.2.0」のソースコードを公開しました。オープンソースなので、無償で誰でも利用できます。「商用利用に耐えるAIを人々に届けたい」、「利用者と共にブラッシュアップしていきたい」という願いでソースコードを公開しました。 【URL】SmallTrain OSSサイト : https://www.smalltrain.org/ 【URL】GitHubリポジトリ: https://github.com/geek-guild/smalltrain


SmallTrainは、ライブラリ機能とラッパー機能を兼ね備えたオープンソースフレームワークです。Kerasのようにつかえます。多様なデータを学習済みのAIモデルを起点に、転移学習をするだけで工数をかけずにAIを構築できます。SmallTrainを使うと、精度を損なわず、大量のデータを取得する労力もいらず、ディープニューラルネットワークを迅速に開発できます。MITライセンスに準拠しており、商用利用できます。





「いち早く世の中に出して、精度の高いAIが低工数でできることをみなさんに体験してもらいたい」という思いから「多くのかたにオープンソースプロジェクトに参加をしてほしい」と完全公開に踏み切りました。ドキュメントを少しずつ増やしていき、プログラムを一から開発しなくてもAIがつくれるようにすることで、AI開発が身近になればと願っています。

今回はAWSのサーバをつかって体験できるチュートリアルを追加しました。AWSのGPUインスタンスの使い方も記載しています。




次のリリースでは、もっと簡単にGoogle Colabでも使えるようにしようと考えています。


SmallTrainにお手持ちのデータを入れて、転移学習をするだけで様々なAIサービスをつくることができます。

SmallTrainの利用事例としては、次のとおりです。

▼ダイナミックプライシング
ホテルの客室価格予測など時系列データを使ってダイナミックプライシングができます。




▼処方箋認識
処方箋を画像認識でデジタル化します。文字を認識します。




▼薬剤画像認識
自然光、薬局の通常のカウンターで薬剤の画像認識ができます。薬剤の種類が多いため、難易度の高い画像認識です。

▼AI-OCR
あらゆるデジタル文字を認識する汎用的な文字認識AIです。
様々なフォーマットに対応できます。

▼画像認識
汎用的な画像認識AIなのであらゆる要望に合致させます。


▼時系列データ予測
発電予測、消費電力予測、発電所異常検知等の、高精度な時系列データ予測をしています。



■SmallTrain 概要

SmallTrainは...
▼主にPythonで記述されており、TensorFlowのラッパーです。将来はKerasやPyTorchのラッパーともなり、TensorFlow、Keras、PyTorchユーザーのかけはしとなることを掲げています。

▼エンジニアやデータサイエンティストに馴染み深いアーキテクチャとして、Pythonモジュールの形式を採用しています。KerasやPyTorchと同じような使い方なので、使い方の学習にかかる時間が短縮され、モデルの構築により多くの時間を費やすことができます。

▼データサイエンスの概念に関する事前の知識がなくても、モジュールを再利用することで、ディープニューラルネットワークを迅速に開発できるように設計されています。

▼オープンソースなので、バグフィックスか改善を開発者1人で抱え込まず、みんなで協力して開発することができます。

▼簡単なインストール

数回クリックするだけで開発できるように、AI学習済みモデルをセットアップできます。

▼高精度モデル

Geek Guildのハイスペックなデータサイエンス手法・GPUマシンを使用してすでにトレーニングした学習済みモデルを公開しています。高精度モデルを起点として、様々な課題解決に最適なソリューションを提案することができます。

▼SmallTrainの使い方・利用シーン

入門として画像認識のモデルの作り方を紹介していますが、時系列データ予測、音声認識など、幅広いAIサービスに使うことができるライブラリ、TensorFlowのラッパーとして使えます。

カテゴリートップへ

本記事はアフィリエイトプログラムによる収益を得ている場合があります

ASCII倶楽部

注目ニュース

  • 角川アスキー総合研究所

プレミアム実機レビュー

ピックアップ
1
KIOXIA(キオクシア) 旧東芝メモリ microSD 128GB UHS-I Class10 (最大読出速度100MB/s) Nintendo Switch動作確認済 国内サポート正規品 メーカー保証5年 KLMEA128G
KIOXIA(キオクシア) 旧東芝メモリ microSD 128GB UHS-I Class10 (最大読出速度100MB/s) Nintendo Switch動作確認済 国内サポート正規品 メーカー保証5年 KLMEA128G
¥2,680
2
Anker PowerLine III Flow USB-C & USB-C ケーブル Anker絡まないケーブル 240W 結束バンド付き USB PD対応 シリコン素材採用 iPhone 17 / 16 / 15 / Galaxy iPad Pro MacBook Pro/Air 各種対応 (1.8m ミッドナイトブラック)
Anker PowerLine III Flow USB-C & USB-C ケーブル Anker絡まないケーブル 240W 結束バンド付き USB PD対応 シリコン素材採用 iPhone 17 / 16 / 15 / Galaxy iPad Pro MacBook Pro/Air 各種対応 (1.8m ミッドナイトブラック)
¥1,890
3
Anker USB Type C ケーブル PowerLine USB-C & USB-A 3.0 ケーブル iPhone 17 / 16 / 15 /Xperia/Galaxy/LG/iPad Pro/MacBook その他 Android 等 USB-C機器対応 テレワーク リモート 在宅勤務 0.9m ホワイト
Anker USB Type C ケーブル PowerLine USB-C & USB-A 3.0 ケーブル iPhone 17 / 16 / 15 /Xperia/Galaxy/LG/iPad Pro/MacBook その他 Android 等 USB-C機器対応 テレワーク リモート 在宅勤務 0.9m ホワイト
¥740
4
Anker iPhone充電ケーブル PowerLine II ライトニングケーブル MFi認証 超高耐久 iPhone 14 / 14 Pro Max / 14 Plus / 13 / 13 Pro / 12 / 11 / X/XS/XR / 8 Plus 各種対応 (0.9m ホワイト)
Anker iPhone充電ケーブル PowerLine II ライトニングケーブル MFi認証 超高耐久 iPhone 14 / 14 Pro Max / 14 Plus / 13 / 13 Pro / 12 / 11 / X/XS/XR / 8 Plus 各種対応 (0.9m ホワイト)
¥990
5
KIOXIA(キオクシア)【日本製】USBフラッシュメモリ 32GB USB2.0 国内サポート正規品 KLU202A032GL
KIOXIA(キオクシア)【日本製】USBフラッシュメモリ 32GB USB2.0 国内サポート正規品 KLU202A032GL
¥1,677
6
エレコム 電源タップ 6個口 3m 雷ガード 個別スイッチ ほこりシャッター付 耐熱 PSE技術基準適合 ブラック T-K6A-2630BK
エレコム 電源タップ 6個口 3m 雷ガード 個別スイッチ ほこりシャッター付 耐熱 PSE技術基準適合 ブラック T-K6A-2630BK
¥1,899
7
バッファロー マウス 無線 ワイヤレス 5ボタン 【戻る/進むボタン搭載】 小型 軽量 節電モデル 最大584日使用可能 BlueLED ブラック BSMBW315BK
バッファロー マウス 無線 ワイヤレス 5ボタン 【戻る/進むボタン搭載】 小型 軽量 節電モデル 最大584日使用可能 BlueLED ブラック BSMBW315BK
¥1,040
8
UGREEN USB Type Cケーブル PD対応 100W/5A 超急速充電 USB C ナイロン編み 断線防止 iphone17/16/15シリーズ/iPad/MacBook Pro/Galaxy S24/Matebook/iPad/Xperia等USB-C各種対応(1m, ブラック)
UGREEN USB Type Cケーブル PD対応 100W/5A 超急速充電 USB C ナイロン編み 断線防止 iphone17/16/15シリーズ/iPad/MacBook Pro/Galaxy S24/Matebook/iPad/Xperia等USB-C各種対応(1m, ブラック)
¥844
9
NIMASO ガラスフィルム iPad 第11世代(A16) 2025用/iPad 10.9インチ 第10世代 2022用 衝撃吸収 強化 ガラス 保護フィルム 指紋防止 ガイド枠付き NTB22I574
NIMASO ガラスフィルム iPad 第11世代(A16) 2025用/iPad 10.9インチ 第10世代 2022用 衝撃吸収 強化 ガラス 保護フィルム 指紋防止 ガイド枠付き NTB22I574
¥1,599
10
KIOXIA(キオクシア)【日本製】SDカード 128GB SDXC UHS-I Class10 読出速度100MB/s 国内正規品 メーカー保証5年 KLNEA128G
KIOXIA(キオクシア)【日本製】SDカード 128GB SDXC UHS-I Class10 読出速度100MB/s 国内正規品 メーカー保証5年 KLNEA128G
¥3,140

Amazonのアソシエイトとして、ASCII.jpは適格販売により収入を得ています。

デジタル用語辞典

ASCII.jpメール デジタルMac/iPodマガジン