-
-
Notifications
You must be signed in to change notification settings - Fork 466
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* yolo object detection initial commit 🎉 * yolo object detection basketball model * visualizations and download scripts * more scripts
- Loading branch information
Showing
8 changed files
with
73 additions
and
0 deletions.
There are no files selected for viewing
10 changes: 10 additions & 0 deletions
10
02_data_science_toolkit/02_yolo_object_detection/download_basketball_model.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
10 changes: 10 additions & 0 deletions
10
02_data_science_toolkit/02_yolo_object_detection/download_chess_dataset.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
10 changes: 10 additions & 0 deletions
10
02_data_science_toolkit/02_yolo_object_detection/download_chess_model.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
10 changes: 10 additions & 0 deletions
10
02_data_science_toolkit/02_yolo_object_detection/download_rolls_model.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 |
33 changes: 33 additions & 0 deletions
33
02_data_science_toolkit/02_yolo_object_detection/setup_yolo_framework.sh
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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.)" |
Binary file added
BIN
+21 MB
...toolkit/02_yolo_object_detection/visualizations/object_detection_basketball.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added
BIN
+14.5 MB
...ence_toolkit/02_yolo_object_detection/visualizations/object_detection_chess.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
Binary file added
BIN
+24.6 MB
...ence_toolkit/02_yolo_object_detection/visualizations/object_detection_rolls.gif
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.