Skip to content
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

Added yolov7 into the sahi model.py #1

Open
wants to merge 17 commits into
base: inference
Choose a base branch
from
Open

Added yolov7 into the sahi model.py #1

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
b9e46db
Added yolov7 package into the sahi model.py
kimb3rlyn Sep 12, 2022
b8a6911
Exposed more parameters of yolov7DetectionModel
kimb3rlyn Sep 15, 2022
3fbf5e6
Tested with other functions of SAHI
kimb3rlyn Sep 19, 2022
f42893c
Removed the rest of the non yolov7 models
kimb3rlyn Sep 20, 2022
2fce9de
Formatting of code
kimb3rlyn Sep 20, 2022
90a29f4
Removing same_size parameter in model
kimb3rlyn Sep 20, 2022
9caeb6b
Changed to pathlib
kimb3rlyn Sep 20, 2022
ab57c55
Edit code in bb to remove extra
kimb3rlyn Sep 20, 2022
565c23c
Format import portion
kimb3rlyn Sep 20, 2022
aa60777
Format import prtion
kimb3rlyn Sep 20, 2022
70bdc4d
Reformat import portion
kimb3rlyn Sep 20, 2022
f457fc9
Format code
kimb3rlyn Sep 20, 2022
ba8a885
Added pull request changes
kimb3rlyn Sep 22, 2022
3d2bff9
Update requirement file
kimb3rlyn Sep 22, 2022
7f2da94
Update Readme.md
kimb3rlyn Sep 22, 2022
bfb7d8c
Change name casing
kimb3rlyn Sep 22, 2022
fafbf8f
Merge branch 'sahi' of github.com:yhsmiley/yolov7 into sahi
kimb3rlyn Sep 22, 2022
File filter

Filter by extension

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
Tested with other functions of SAHI
  • Loading branch information
@kimb3rlyn
kimb3rlyn committed Sep 19, 2022
commit 3fbf5e626541dbf6144af28d271ac1b9f25562ad
2 changes: 0 additions & 2 deletions scripts/inferenceSahi.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,6 @@
from importlib import resources
from sahi.predict import predict,get_prediction,get_sliced_prediction
from yolov7.utils.sahiModelExtended import Yolov7DetectionModel
from yolov7.utils.general import increment_path
from sahi.slicing import slice_image

# same_size : True if all the image that are passed in are the same size (defaulted to True)
Expand All @@ -17,7 +16,6 @@
weightPath = resources.files('yolov7').joinpath('weights/yolov7-e6_state.pt')
cfgPath = resources.files('yolov7').joinpath('cfg/deploy/yolov7-e6.yaml')


model = Yolov7DetectionModel(
image_size = 1280,
model_path = weightPath,
Expand Down