Open koles289 opened 4 years ago
Hi, I am quite new to tensorflow and I would like to fine-tune _centernet_hg104_512x512_coco17tpu-8 model from modelzoo on my own images. I was following this tutorial: https://github.com/tensorflow/models/blob/master/research/object_detection/colab_tutorials/eager_few_shot_od_training_tf2_colab.ipynb. For SSD training is used attribute _boxpredictor, but centernet does not have this attribute.
I would like to ask, how to select trainable layers in Centernet. From model_builder.py I was not able to solve this question.
I would be grateful for any help
I have the same problem. Also, I would like to fine-tune EfficientDet in Colab too.
Hi, I am quite new to tensorflow and I would like to fine-tune _centernet_hg104_512x512_coco17tpu-8 model from modelzoo on my own images. I was following this tutorial: https://github.com/tensorflow/models/blob/master/research/object_detection/colab_tutorials/eager_few_shot_od_training_tf2_colab.ipynb. For SSD training is used attribute _boxpredictor, but centernet does not have this attribute.
I would like to ask, how to select trainable layers in Centernet. From model_builder.py I was not able to solve this question.
I would be grateful for any help