search

tensorflow / models

Models and examples built with TensorFlow
Other
76.91k stars 45.81k forks source link

training Centermodel - centernet_hg104_512x512_coco17_tpu-8 #9011

Open koles289 opened 4 years ago

koles289 commented 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

DanilKonon commented 3 years ago

I have the same problem. Also, I would like to fine-tune EfficientDet in Colab too.