fregu856/2d_detection

TensorFlow implementation of SqueezeDet, trained on the KITTI dataset.

2d_detection's Issues

Train SqueezeDet for own dataset for 1 class

Hi,
How can I do transfer learning or fine-tuning the pretrained SqueezeDet model for own dataset for 1 class??

Nothing output

With python2.7, or 3.5, the code can not output anything. Could you know the reason?

Frozen version of the net

Hi,
Thanks for your great implementation.
I would be interested to get a "frozen" version of the net.

Could you add to the repository the frozen version of the graph, please ?
You can take a look here :
https://github.com/tensorflow/tensorflow/blob/r1.8/tensorflow/python/tools/freeze_graph.py
or
https://github.com/tensorflow/models/tree/master/research/slim#freezing-the-exported-graph

Thank you in advance

how to load the model

thanks for sharing the project.
where is the script to load the model to detect the pic ?

utilities.py", line 40, in draw_bboxes

2D_detection/utilities.py", line 40, in draw_bboxes
for bbox, class_label, prob in zip(bboxes, class_labels, probs):
TypeError: zip argument #3 must support iteration

Don't know how to solve it?

Package version problem?