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.

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

Initial support for YOLOv3-tiny and YOLOv5 #3099

Closed
wants to merge 6 commits into from

Conversation

triptec
Copy link

@triptec triptec commented Apr 13, 2022

This PR builds on the hard work of @no2chem's PR and @jveitchmichaelis's repo and @iaverypadberg's fork, amazing work!

My contribution is mainly copy+paste but it seems someone needs to do that as well ;)

Now I'm not suggesting this be merged. I just really wanted to be able to detect cats pissing in my carport and yolov5 does that for me without too many false positives. Now I'm kind of hoping this will inspire someone with more time/energy on their hands to clean this mess up and get it into frigate =)

To try this branch out

  1. Follow the setup instructions
  2. Get one of the models from here and labels from here
  3. And update the config accordingly, for example:
mqtt:
  host: mqtt

cameras:
  test:
    ffmpeg:
      inputs:
        - path: /media/frigate/cat1-15.MP4
          input_args: -re -stream_loop -1 -fflags +genpts
          roles:
            - detect
            - rtmp
    detect:
      width: 1920 # <---- update for your camera's resolution
      height: 1080 # <---- update for your camera's resolution

motion:
  # Optional: The threshold passed to cv2.threshold to determine if a pixel is different enough to be counted as motion. (default: shown below)
  # Increasing this value will make motion detection less sensitive and decreasing it will make motion detection more sensitive.
  # The value should be between 1 and 255.
  threshold: 25

model:
  path: /media/frigate/yolov5s-int8-224.tflite
  height: 224
  width: 224
  type: 'yolov5'
  labelmap_path: '/media/frigate/coco_labels.txt'

@netlify
Copy link

netlify bot commented Apr 13, 2022

Deploy Preview for frigate-docs ready!

Name Link
🔨 Latest commit b7429d3
🔍 Latest deploy log https://app.netlify.com/sites/frigate-docs/deploys/6257228178192900086b12fc
😎 Deploy Preview https://deploy-preview-3099--frigate-docs.netlify.app
📱 Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site settings.

@NickM-27
Copy link
Collaborator

For it to be mergeable it would need to target the release branch, not master.

@triptec
Copy link
Author

triptec commented Apr 13, 2022

@NickM-27 against the release branch? It seems both ahead and behind master?

@NickM-27
Copy link
Collaborator

NickM-27 commented Apr 13, 2022

@NickM-27 against the release branch? It seems both ahead and behind master?

Yes, new features go to the release branch. Docs updates and bug fixes for current release go to master. (This is current branching strategy, blake said he'll revisit after another contributor discussed potential changes here: #3081)

@triptec triptec changed the base branch from master to release-0.11.0 April 13, 2022 19:21
@blakeblackshear
Copy link
Owner

There are a lot of changes to add support for these model architectures. Are the models you are using trained on the same COCO dataset as all the other models? What benefits are you seeing?

@triptec
Copy link
Author

triptec commented Apr 15, 2022 via email

@blakeblackshear
Copy link
Owner

You either have to screen record or use the mjpeg feed as an input to ffmpeg.

@triptec
Copy link
Author

triptec commented Apr 21, 2022

Ok, I'll give that a try. I've also just received a coral dev board and thought I should try this code on it.

@stale
Copy link

stale bot commented Jun 18, 2022

This issue has been automatically marked as stale because it has not had recent activity. It will be closed if no further activity occurs. Thank you for your contributions.

@stale stale bot added the stale label Jun 18, 2022
@stale stale bot closed this Jun 22, 2022
@triptec triptec mentioned this pull request Nov 26, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants