Skip to content

C++ Demo - Object Tracking (VitTrack) #240

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.

Sign up for GitHub

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

Merged
merged 8 commits into from
Mar 6, 2024
Merged

C++ Demo - Object Tracking (VitTrack) #240

Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
6e73aa5
Preliminary attempt at C++ demo.
ryan1288 Mar 2, 2024
ad19119
Update README documentation
ryan1288 Mar 2, 2024
8a10724
Fixed text spacing and removed unused arguments. Cleaned up to not us…
ryan1288 Mar 3, 2024
7b8c179
Update offsets to match C++ to prevent overlapping text
ryan1288 Mar 3, 2024
cb6d08e
Add help functionality
ryan1288 Mar 3, 2024
e19f388
Add using namespace for standalone C++ demo file for readability.
ryan1288 Mar 3, 2024
b3d0cd3
Update formatting and add save/visualization functionality
ryan1288 Mar 5, 2024
11beab7
More formatting changes
ryan1288 Mar 5, 2024
File filter

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add help functionality
  • Loading branch information
@ryan1288
ryan1288 committed Mar 3, 2024
commit cb6d08ed5db2df6db290cd47c5990f36e92e4416
6 changes: 6 additions & 0 deletions models/object_tracking_vittrack/demo.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -60,11 +60,17 @@ cv::Mat visualize(const cv::Mat& image, const cv::Rect& bbox, float score, bool

int main(int argc, char** argv) {
cv::CommandLineParser parser(argc, argv,
"{help h | | Print help message. }"
"{input i | |Set path to the input video. Omit for using default camera.}"
"{model_path |object_tracking_vittrack_2023sep.onnx |Set model path}"
"{backend_target bt |0 |Choose backend-target pair: 0 - OpenCV implementation + CPU, 1 - CUDA + GPU (CUDA), 2 - CUDA + GPU (CUDA FP16), 3 - TIM-VX + NPU, 4 - CANN + NPU}"
"{save s |false |Specify to save a file with results. Invalid in case of camera input.}"
"{vis v |false |Specify to open a new window to show results. Invalid in case of camera input.}");
if (parser.has("help"))
{
parser.printMessage();
return 0;
}

std::string input_path = parser.get<std::string>("input");
std::string model_path = parser.get<std::string>("model_path");
Expand Down
1 change: 0 additions & 1 deletion models/object_tracking_vittrack/demo.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,6 @@
parser.add_argument('--vis', '-v', action='store_true',
help='Usage: Specify to open a new window to show results. Invalid in case of camera input.')
args = parser.parse_args()

def visualize(image, bbox, score, isLocated, fps=None, box_color=(0, 255, 0),text_color=(0, 255, 0), fontScale = 1, fontSize = 1):
output = image.copy()
h, w, _ = output.shape
Expand Down