1-stage-wsegのバグとかエラーの殴り書き

github.com

をやっとる。

・pascal_vocのみに対応

・pascal_voc.pyで学習データのフォーマットとかクラス数とかクラス名を定義しているっぽい。def __getitem__でクラスを取得している。

・backboneをresnet101にするときはvoc_resnet101.yamlの「LR_SCHEDULER: "none"」を消さないとエラーになる

・READMEには書いてないけど、画像のファイル名を書いたtrain_aug.txtが必要っぽい