tensorflow / tensorflow/models
How to detect objects in tensor flow for specific objects in video for example only bottle
Open
@pkulzc is already working on this.
Since May 29, 2020.
models:research:odapi
type:support
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
to detect objects in tensor flow for specific objects in the video for example bottle.
I have indices = np.argwhere(classes == 44)* where class 44 is for bottle and it is throwing an error :too many indices for array
and it is working fine for class 1 which is for persons. it is working fine
Can someone help me with this?
Visualization of the results of detection.
boxes = np.squeeze(boxes)
scores = np.squeeze(scores)
classes = np.squeeze(classes)
**indices = np.argwhere(classes == 44)**
boxes = np.squeeze(boxes[indices])
scores = np.squeeze(scores[indices])
classes = np.squeeze(classes[indices])
vis_util.visualize_boxes_and_labels_on_image_array(
image_np,
np.squeeze(boxes),
np.squeeze(classes).astype(np.int32),
np.squeeze(scores),
category_index,
use_normalized_coordinates=True,
line_thickness=8)
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.