yuuho.wiki

カオスの欠片を集めて知恵の泉を作る

ユーザ用ツール

サイト用ツール


tips:python:detectron2:start

差分

このページの2つのバージョン間の差分を表示します。

この比較画面へのリンク

両方とも前のリビジョン前のリビジョン
次のリビジョン
前のリビジョン
tips:python:detectron2:start [2022/01/31 09:27] – [自作データセット定義方法] yuuhotips:python:detectron2:start [2022/02/01 01:37] (現在) – [自作データセット定義方法] yuuho
行 242: 行 242:
  
 </code> </code>
 +
 +=== カスタムなデータローダーの定義 ===
 +
 +デフォルトでは ''detectron2/data/dataset_mapper.py'' の ''DatasetMapper'' が使われるため画像はファイルから読み込むしかできない。
 +データセットに合った特定の読み込み方を指定するには custom dataloader を作成する必要がある。
 +  * [[https://github.com/facebookresearch/detectron2/blob/cbbc1ce26473cb2a5cc8f58e8ada9ae14cb41052/docs/tutorials/datasets.md|custom dataset]]
 +  * [[https://github.com/facebookresearch/detectron2/blob/cbbc1ce26473cb2a5cc8f58e8ada9ae14cb41052/docs/tutorials/data_loading.md|custom data loader]]
 +
 +これらを見る限り、 ''DefaultTrainer'' の ''build_train_loader'' などのメソッドをオーバーライドするのが正攻法。
 +
tips/python/detectron2/start.1643621265.txt.gz · 最終更新: 2022/01/31 09:27 by yuuho