-
-
Notifications
You must be signed in to change notification settings - Fork 265
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
gpu support for autolabeling is now possible for yolov5 #4
Conversation
opencv_dnn.sh
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think this file is not necessary because it cannot cover well on different machines.
Please remove this file. We will create a documentation site in the future for installation guide.
opencv_dnn.sh
Outdated
|
||
|
||
# ln -s command is used to create a symbolic link, also known as a symlink or soft link | ||
cd ~/.virtualenvs/opencv_cuda/lib/python3.8/site-packages/ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Maybe other users do not have ~/.virtualenvs
as you.
opencv_dnn.sh
Outdated
# Uninstall openCV dependencies, since dnn module installed from source is not recognize by pip | ||
pip uninstall opencv-contrib-python-headless opencv-python-headless | ||
|
||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You should not add too many new lines in the end.
1 is enough.
@@ -167,7 +167,7 @@ def save( | |||
if flags is None: | |||
flags = {} | |||
data = dict( | |||
version=__version__, | |||
#version=__version__, |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please merge with the latest master. I have fixed this issue. Thanks.
If dnn module with GPU support for openCV is available, the labeling will be done into the GPU.
You must install your dnn module with GPU support following by running the file
opencv_dnn.sh
.This file is based on the instructions provided by Adrian Rosebrock (@jrosebr1) and he has the whole credit and not me.
TODO List