COVIDNet-CT icon indicating copy to clipboard operation
COVIDNet-CT copied to clipboard

Published 20 hours ago • verified publisher icon haydengunraj

Model implementation

Open hoangtnm opened this issue 2 years ago • 1 comments

Hi, I'm really interested in COVID-Net CT-2 models, I tried to search through the repo but I couldn't find the model implementation in Python (TensorFlow) so that I can reproduce them with PyTorch. Can you share the original implementation of such models? thank you

hoangtnm avatar Oct 03 '21 01:10 hoangtnm

Hi Tran, sorry for the late response. Unfortunately these models were constructed using a machine-driven design strategy in which the original model is modified automatically. The downside of this approach is that it produces a TensorFlow meta graph file for the modified model, but not the source code for constructing the modified model.

To convert the model to PyTorch, your best bet would probably be to view the graph in TensorBoard - this allows you to see the layers, connections, input/output shapes, channel counts, etc. There is also a somewhat high-level diagram of the architecture in the CT-2 paper which can help with the general structure, and this diagram also includes the channel counts for the various layers.

haydengunraj avatar Apr 26 '22 22:04 haydengunraj