search

nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[FEATURE] Refactor new model registration #24

Closed alphinside closed 4 years ago

alphinside commented 4 years ago

Describe the solution you'd like Currently new model is registered by modify init.py. To make it more flexible, model registration may utlize python decorator so it no longer need to modify init.py

Also if possible, custom model registration may be handle similar like external dataset