Skip to content

Commit

Permalink
Yolo object detection (#18)
Browse files Browse the repository at this point in the history
* yolo object detection initial commit 🎉

* yolo object detection basketball model

* visualizations and download scripts

* more scripts
  • Loading branch information
SkalskiP committed Jan 30, 2020
1 parent aeb9b3b commit 511be9d
Show file tree
Hide file tree
Showing 8 changed files with 73 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash

# Download pre-trained model along with sample images and config files
curl https://yolo-models.s3.us-east-2.amazonaws.com/basketball.zip -o basketball.zip
unzip basketball.zip
rm basketball.zip

# Move config and data files
cp basketball/data/* yolov3/data
cp basketball/cfg/* yolov3/cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash

# Download pre-trained model along with sample images and config files
curl https://yolo-models.s3.us-east-2.amazonaws.com/chess_dataset.zip -o chess_dataset.zip
unzip chess_dataset.zip
rm chess_dataset.zip

# Move config and data files
cp chess_dataset/data/* yolov3/data
cp chess_dataset/cfg/* yolov3/cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash

# Download pre-trained model along with sample images and config files
curl https://yolo-models.s3.us-east-2.amazonaws.com/chess.zip -o chess.zip
unzip chess.zip
rm chess.zip

# Move config and data files
cp chess/data/* yolov3/data
cp chess/cfg/* yolov3/cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
#!/bin/bash

# Download pre-trained model along with sample images and config files
curl https://yolo-models.s3.us-east-2.amazonaws.com/rolls.zip -o rolls.zip
unzip rolls.zip
rm rolls.zip

# Move config and data files
cp rolls/data/* yolov3/data
cp rolls/cfg/* yolov3/cfg
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
#!/bin/bash

# Clone framework
git clone https://github.com/ultralytics/yolov3.git

# Getting pre-trained weights
cd ./yolov3
sh weights/download_yolov3_weights.sh

# Setup virtual env
sudo apt-get update
sudo apt-get install unzip
sudo apt-get install python-dev
sudo apt-get install python-pip
sudo apt-get install python3-venv
python3 -m venv .env
source .env/bin/activate
pip3 install --upgrade pip
pip3 install --upgrade setuptools
pip3 install Cython
pip3 install numpy
pip3 install "pillow<7"
pip3 install -r requirements.txt
pip3 install torchvision
deactivate

echo "*****************************************************************************"
echo "****** Your environment has been created ******"
echo "*****************************************************************************"
echo ""
echo "If you had no errors, You can proceed to work with your virtualenv as normal."
echo "(run 'source .env/bin/activate' in your assignment directory to load the venv,"
echo " and run 'deactivate' to exit the venv. See assignment handout for details.)"
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.

0 comments on commit 511be9d

Please sign in to comment.