-
Notifications
You must be signed in to change notification settings - Fork 45.3k
Description
I followed this tutorial https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10
I know that by default, TF API initialize from a checkpoint but allow retraining for all layers.
But I just want to retrain only the last layer, because I want to use pre-trained model in COCO to train in Open Images dataset.
Are there anyone can tell me how to do that?
System information
What is the top-level directory of the model you are using:(I don't really understand the meaning of "top-level directory of the model")
I just followed https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10
I used faster_rcnn_nas_lowproposals_coco downloaded from
https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/detection_model_zoo.md
OS Platform and Distribution:
Ubuntu 16.04
TensorFlow installed from:
GPU version of Tensorflow
TensorFlow version:
v1.12
Bazel version:
NA: don't know what is Bazel
CUDA/cuDNN version:
NA: don't know, but it's not the related to the problem
GPU model and memory:
Titan V
Exact command to reproduce:
NA