-
Notifications
You must be signed in to change notification settings - Fork 30
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.
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
What changes to make to run in using our own yolov5 moels? #5
Comments
For DeepSort checkpoint file For Yolov5s weight file To use your own model,
|
Do I need to replace both the deepsort checkpoint and the custom yolov5s.pt file to check results of my custom model? |
First try replacing with your Yolov5 weight file. Most probably it'll get the job done. If, after passing your own Yolov5 weight file, you don't see tracking then I think you will have to see how to train and work with DeepSort checkpoint file. But I think, just by replacing with your Yolov5 file, it will work without any issues. |
Great! One last question. Can we deploy this on AWS and use it there? If yes, do you have any tutorials or any steps? |
If you are thinking to deploy this Streamlit app to AWS, then
I have worked on a similar thing so the above approach should work as per my experience. If not, I think you can check in Streamlit docs. NOTE - There is a bug in Streamlit. When running any Streamlit app in any cloud VM (OR even local, when passing video/image from some different folder), when you pass video/image via file uploader in Streamlit, the code won't be able to read the image/video because Streamlit doesn't capture the path of file uploaded. Solution - download the video/image, pass it via Streamlit file uploader to a temporary folder and then pass this path to the code. |
any docker file instructions to be added in the dockerfile? |
Yeah sure. I will try to add it to this repo by end of the day. |
Sure Thank you |
@SahilChachra Hi, any docker image created? |
@joel5638 Hey. I'm busy with few things for few more days. Sorry for late reply. Will upload the docker file as soon as it's possible. |
Do we have to change the deepsort checkpoint?
We can replace the yolov5s.pt with our custom model. But what else changes do we have to make?
The text was updated successfully, but these errors were encountered: