17 lines
704 B
Python
17 lines
704 B
Python
import openpifpaf
|
|
|
|
from . import GrandPrix_kp
|
|
|
|
|
|
def register():
|
|
openpifpaf.DATAMODULES['GrandPrix'] = GrandPrix_kp.GrandPrix_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"
|