how to train own datasets?

Bisonai / mobilenetv3-tensorflow

Unofficial implementation of MobileNetV3 architecture described in paper Searching for MobileNetV3.

Apache License 2.0

225 stars 81 forks source link

Closed mumuyanyan closed 4 years ago

mumuyanyan commented 4 years ago

there are only two datasets ? how to train own datasets?

martinkersner commented 4 years ago

Hi @mumuyanyan ! If you want to use other datasets you could just past different name to build_dataset function and get any of Tensorflow datasets https://github.com/Bisonai/mobilenetv3-tensorflow/blob/master/datasets.py#L32