tensorflow / tensorflow/models
Trained model detects other objects
Open
@pkulzc is already working on this.
Since Jun 23, 2020.
models:research:odapi
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
Hello! I have trained my model using /object_detection/model_main.py with SSD_mobilenet_v1_coco model. I have trained it to detect glasses for something like ~20,000 steps, after the training it detects glasses with 99-100% accuracy. But it also detects everything, that usual mobilenet v1 coco model detect(around 90 other objects) and labels them as glasses. What i must do to make my model stop detecting other objects?
System information
- What is the top-level directory of the model you are using: trained model
- Have I written custom code (as opposed to using a stock example script provided in TensorFlow): no
- OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Ubuntu 16.04
- TensorFlow installed from (source or binary): source
- TensorFlow version (use command below): tensorflow-gpu
- Bazel version (if compiling from source): n/a
- CUDA/cuDNN version: 10.1
- GPU model and memory: 2080ti 11264 MB
- Exact command to reproduce: n/a
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.