Skip to content

Model and test data for the RMRR deepstack model, primarily for detecting Texas wildlife

License

Notifications You must be signed in to change notification settings

avatar42/RMRR.model

Repository files navigation

RMRR DeepStack model RMRR

Model and test data for the RMRR (Round Mountain Rescue Ranch) DeepStack model. Primarily an proof of concept of how to quickly create a model from a collection of images from your security cams. See my DeepStack Utils repo Its main purpose is the ID the critters running around my place.

See Releases Info for more detail about the model's creation and evolution.

As of v0.4 the train and test folders with the image and map files have been moved to Kaggle. Download them as zip files from here

How to use:

To use the model download the RMRR.pt file to your C:\DeepStack\MyModels folder. (Can be any folder but Blue Iris defaults to C:\DeepStack\MyModels)

  1. Add a folder inside your DeepStack instance (If you run on Host) or mount it in Docker.

  2. Run DeepStack with the command --MODELSTORE-DETECTION "C:/DeepStack/MyModels" See DeepStack utils for more info on setting up and testing your set up.

  3. To use RMRR detection call the end point /v1/vision/custom/RMRR with your picture and you get a response similar to this if it finds a known object:

{
'success': True,
'predictions': [{
  'confidence': 0.93365675,
  'label': 'raccoon',
  'y_min': 279,
  'x_min': 640,
  'y_max': 340,
  'x_max': 767
  }]
}

The list of currently trained for objects in in RMRR_classes.txt

trainTest results are in train.trainTest.results.txt and test.trainTest.results.txt

For more details on training see DeepStack training

For more help setting and or debugging a DeepStack setup see Quick Blue Iris with DeepStack debug

See runTrain.bat for an example of running training locally with my DeepStack Utils repo style setup.

Training results

confusion_matrix.png labels.jpg labels_correlogram.jpg precision_recall_curve.png results.png test_batch0_labels.jpg test_batch0_pred.jpg test_batch1_labels.jpg test_batch1_pred.jpg test_batch2_labels.jpg test_batch2_pred.jpg train_batch0.jpg train_batch1.jpg train_batch2.jpg

About

Model and test data for the RMRR deepstack model, primarily for detecting Texas wildlife

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published