Skip to content

USC CSCI596 Scientific Computing and Visualization Final Project

Notifications You must be signed in to change notification settings

maplezzz/CSCI596FinalProject

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

CSCI596FinalProject

USC CSCI596 Scientific Computing and Visualization Final Project

Parallel Computing based on pytroch and CUDA to implement YOLO V3

you can find the detail about YOLO V3 here

The architecture of YOLOV3 like that
archi
this project will only focuse on reference not training, this program will work on GPU if you have installed pytorch with cuda

How to use

Requirements

  • Python >= 3.5
  • OpenCV
  • PyTorch >= 0.4

the weights of the yolov3 is large, so you can download here, then put it in root

you can store the images you want to be dectected in imgs file.
run the command

python detect.py --images imgs --det det 

you will get the result in detect file

Result

image detection, you can find the raw image in imgs file dog 1 2 3 person

About

USC CSCI596 Scientific Computing and Visualization Final Project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages