17 lines
700 B
Python
17 lines
700 B
Python
import openpifpaf
|
|
|
|
from . import formulaE_kp
|
|
|
|
|
|
def register():
|
|
openpifpaf.DATAMODULES['formulaE'] = formulaE_kp.FormulaE_Kp
|
|
openpifpaf.CHECKPOINT_URLS['shufflenetv2k16-apollo-24'] = \
|
|
"http://github.com/DuncanZauss/openpifpaf_assets/releases/" \
|
|
"download/v0.1.0/shufflenetv2k16-201113-135121-apollo.pkl.epoch290"
|
|
openpifpaf.CHECKPOINT_URLS['shufflenetv2k16-apollo-66'] = \
|
|
"http://github.com/DuncanZauss/openpifpaf_assets/releases/" \
|
|
"download/v0.1.0/sk16_apollo_66kp.pkl"
|
|
openpifpaf.CHECKPOINT_URLS['shufflenetv2k30-apollo-66'] = \
|
|
"http://github.com/DuncanZauss/openpifpaf_assets/releases/" \
|
|
"download/v0.1.0/sk30_apollo_66kp.pkl"
|