Comments (4)
you can change the GPU Flag in YOLO :
FLAGS.gpuName = '/gpu:0'
by default it's on GPU 0 , but you can set it to another GPU (for example /gpu:1) , hopefully deep_sort will still run on GPU 0 (I only have one GPU so can't test it )
Another thing to try is to set deep sort to run on GPU 1 , by adding this line to deep sort tf_session :
config = tf.ConfigProto(device_count = {'GPU': 1})
session = tf.Session(config=config)
these should replace line 318 in the script deep_sort/generate_detection.py
if it works out, a pull request would be appreciated for future users :)
from tracking-with-darkflow.
I tried the two ways you mentioned, however it both turns out that they have no effects on the gpu assignment.
it always used both of the memory of gpu0 and gpu1
but the gpu util shows only the percent value(like 40%-60%)on gpu0, while 0% on gpu 1
from tracking-with-darkflow.
thanks for reply!
i will try to test it, hopefully it works.
from tracking-with-darkflow.
try using with tf.device("GPU:1"):
before calling the encoder
I'm sorry I don't have the required hardware so can't really help much
from tracking-with-darkflow.
Related Issues (20)
- How can I get direction of Individual ID's?
- ModuleNotFound Error : No module named 'darkflow.cython_utils' while setting up on Google Colab
- Video is not getting saved
- When I run the code "python run.py" ,there is an error . how can I solve it? HOT 1
- the deep_sort weights provided by nwojke is cancelled HOT 8
- ssertionError: Cannot capture source HOT 1
- Not detecting people HOT 1
- darkflow
- Why is the target size different from that trained by Darknet after using this code?
- Hello, I use yolov2 trained by darknet and mark the head and shoulders of the person, but when I use the trained model in this code, I can only detect the head, not the head and shoulders. That is why ah! Does anyone have the same problem with me?
- doesn't save the video HOT 1
- Hello, I used darkflow training to generate four files, but how do I use it in the code?
- I trained yolov2 under darknet. The effect is not good when using this code. Why is this?
- Excuse me? Can this code use pb files? HOT 2
- OOM error
- The name 'net/images:0' refers to a Tensor which does not exist
- AVAssetWriter status: Cannot create file HOT 3
- Using Mask-RCNN for detection HOT 1
- What would be the guideline for multi camera tracking ?
- GPU memory usage 9G
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
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.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from tracking-with-darkflow.