How to train my task

[lilitong7]

Hi, i'd like to train my dataset with the yoloface-50k model, but i dont find any pytorch or tensorflow code for training.Can you tell me where i can find that if you know one. Thank you!

[imuncle]

The original repo of this model is https://github.com/dog-qiuqiu/MobileNet-Yolo/tree/master/yoloface-50k.

This model is build based on the original Yolo, which is written in C. So no pytorch or tensorflow code.

If you want to train your own dataset, you can use part of my pytorch code https://github.com/imuncle/yoloface-50k/blob/main/pytorch/yoloface.py. In this code, I convert the network into pytorch. Then you can add your own train code.

[lilitong7]

The original repo of this model is https://github.com/dog-qiuqiu/MobileNet-Yolo/tree/master/yoloface-50k.

This model is build based on the original Yolo, which is written in C. So no pytorch or tensorflow code.

If you want to train your own dataset, you can use part of my pytorch code https://github.com/imuncle/yoloface-50k/blob/main/pytorch/yoloface.py. In this code, I convert the network into pytorch. Then you can add your own train code.

Thanks for your replys.I do just as what you say, and export model as .onnx data successfully. However, i'd like to analyze it by Cube-AI and there is an error "[AI:network] NOT IMPLEMENTED: 4 or more number of tensor input dimensions are not implemented". Then, i put tflite model of yoloface-500k model into Cube-AI and there is no wrong. Therefore, i wonder if pytorch model cant be analyzed by Cube-AI cuz tensor in torch is like [B,C,H,W] and tensor in tensorflow is like [B,H,W,C].i sincerely appreciate it if you can give me some opinions.