Skip to content

guome/RepVGG-Tensorflow-2

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RepVGG: Making VGG-style ConvNets Great Again (Tensorflow 2)

TODO: Usage

TODO: Convert the training-time models into inference-time

TODO: ImageNet training

FAQs

Q: Is the inference-time model's output the same as the training-time model?

A: Yes. You can verify that by

import tensorflow as tf
import numpy as np
train_model = create_RepVGG_A0(deploy=False)
train_model.build(input_shape=(None, 64, 64, 3))
deploy_model = repvgg_model_convert(train_model, create_RepVGG_A0, image_size=(64, 64, 3))
x = tf.random.uniform((32, 64, 64, 3))
train_y = train_model(x)
deploy_y = deploy_model(x)
print(np.mean((train_y - deploy_y) ** 2))    # Will be around 1e-10

Related works

Reference

Author

HOANG Duc Thang

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 96.0%
  • Dockerfile 4.0%