thtrieu/darkflow

How to save features of each detected bounding box?

Open

#65 opened on Feb 22, 2017

View on GitHub
 (3 comments) (4 reactions) (0 assignees)Python (6,120 stars) (2,102 forks)batch import
help wanted

Description

I would like to save the features of each detected bounding box for later use. How can this be done? How can I select the specific layer from which I want the features to be extracted? Thank you

Contributor guide

Tech stack
pythontensorflow
Domain
machine learningai
Issue type
research
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
under 1 hour
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
stale
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
clear
Prerequisites
basic PythonTensorFlow basicsobject detection concepts
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
40
Research direction
To extract features from a specific layer, you need to modify the inference script (e.g., darkflow/net/flow.py) to intercept the output of a given layer. Identify the layer name from the YOLO architecture (e.g., 'conv 23') and add a TensorFlow hook using tf.Graph.get tensor by name(). Then, feed the input image and retrieve the tensor values for each bounding box. Check existing comments in the issue for hints on implementation.