Giter VIP home page Giter VIP logo

Comments (3)

mustafa-qamaruddin avatar mustafa-qamaruddin commented on June 16, 2024

That's how I freeze the graph:

$ python tools/freeze_graph.py \
    --input_graph /tmp/mobilenet-model-0004/graph.pbtxt \
    --input_checkpoint /tmp/mobilenet-model-0004/model.ckpt-0 \
    --output_graph ./frozen_graph.pb \
    --output_node_names MobileNet/Predictions/Softmax

That's how I did the training:

$ python train_image_classifier.py \
    --train_dir=/tmp/mobilenet-model-0004 \
    --dataset_name=custom \
    --dataset_split_name=train_val \
    --dataset_dir=LEGOSET/records \
    --model_name=mobilenet \
    --preprocessing_name=mobilenet \
    --width_multiplier=1.0 \
    --max_number_of_steps=1000 \
    --batch_size=32 \
    --save_interval_secs=240 \
    --save_summaries_secs=240 \
    --log_every_n_steps=100 \
    --optimizer=adam \
    --opt_epsilon=0.000000001 \
    --learning_rate=0.0001 \
    --learning_rate_decay_factor=0.00001 \
    --num_epochs_per_decay=2 \
    --weight_decay=0.00005 \
    --adam_beta1=0.09 \
    --adam_beta2=0.0999

from mobilenet.

mustafa-qamaruddin avatar mustafa-qamaruddin commented on June 16, 2024

input_node = graph.get_tensor_by_name('import/MobileNet/input_images:0')
gives the following:
KeyError: "The name 'import/MobileNet/input_images:0' refers to a Tensor which does not exist. The operation, 'import/MobileNet/input_images', does not exist in the graph."

from mobilenet.

DRAWPIETY avatar DRAWPIETY commented on June 16, 2024

Hello, I have the same mistake as you. Can you tell me how to solve it?

from mobilenet.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.