/Ultra-Light-Fast-Generic-Face-Detector_Tensorflow-Model-Converter

a tool for convert Ultra-Light-Fast-Generic-Face-Detector from pytorch to tensorflow 2.0+

Primary LanguagePython

Ultra-Light-Fast-Generic-Face-Detector-Tensorflow Converter

This project is base on Ultra-Light-Fast-Generic-Face-Detector-1MB. You can use this script to converter origin model to tensorflow version.

Running Environment

  • tensorflow 2.3.1
  • torch 1.6.0
  • opencv-python 4.4.0.44

Run

Covert model

 python3 ./convert_tensorflow.py --net_type <RFB|slim>

Inference on image

 python3 ./det_image.py --net_type <RFB|slim> --img_path <path>

Inference on Webcam

 python3 ./det_webcam.py --net_type <RFB|slim>

Result Comparison

Pytorch Slim

img1

Tensorflow Slim

img1

Pytorch RFB

img1

Tensorflow RFB

img1

Reference