aifarms / multi-camera-pig-tracking Goto Github PK

Official Implementation of "Tracking Grow-Finish Pigs Across Large Pens Using Multiple Cameras"

Home Page: https://www.youtube.com/watch?v=wFGXfTkzE3E&t=2213s

Python 11.10% Makefile 0.22% C 83.21% Cuda 5.24% C++ 0.23%

multi-camera-tracking pig-detection pig-tracking homography computer-vision yolov4 deepsort deep-learning multi-object-tracking dataset

multi-camera-pig-tracking's Introduction

Multi Camera Pig Tracking

Official Implementation of Tracking Grow-Finish Pigs Across Large Pens Using Multiple Cameras

CVPR2021 CV4Animals Workshop Poster

Dataset

The dataset can be found at this link.

The videos were acquired at the Imported Swine Research Lab (ISRL) at UIUC. The deployment video can be found here. It was annotated for grountruth global identities and bounding boxes using this MATLAB Tool.

Files

deepsort-tracking: Contains code for detecting and tracking pigs using YOLOv4 and DeepSORT

data/homography: Contains pickled homography matrices for both the pens

camera.py: Detects and tracks pigs using the trained model in DeepSORT and YOLOv4

manager.py: Main file which uses the homography matrices to assign global identities

Running the code

Download the dataset from this link and place multicam-dataset folder in data/. Note that we have already trained the model and extracted the output of DeepSORT into JSON files. You can find the pretrained checkpoint here

Run export DARKNET_PATH=./deepsort-tracking/yolov4/ in terminal.
Run any one of the commands from commands.txt, for instance: python3 manager.py --av data/multicam-dataset/0/0-Pen_B.mp4 --cv data/multicam-dataset/0/0-Ceiling_Cam.mp4 --cj data/multicam-dataset/0/0-Ceiling_Cam.json --aj data/multicam-dataset/0/0-Pen_B.json --cl 457

Future Work

We are currently working on building action recognition models for pig behavior using ethograms. We can currently estimate the time spent by pigs near drinkers and feeders based on their proximity.

multi-camera-pig-tracking's People

Contributors

Stargazers

Watchers

Forkers

ayush1801 em-biet-code saulocatharino araneto slkang2001 khangdong89 chishengchen imhgchoi jacknion zhangshaopu trongthanhqvo brahimmade narsaskpvt mamun-illini andreemidio durgaprasadmamidi swineologist ngobibibnbe

multi-camera-pig-tracking's Issues

Error in running manager.py

Hi
I am following the steps to running the code .I found missing packages
Run any one of the commands from commands.txt, for instance: python3 manager.py --av data/multicam-dataset/0/0-Pen_B.mp4 --cv data/multicam-dataset/0/0-Ceiling_Cam.mp4 --cj data/multicam-dataset/0/0-Ceiling_Cam.json --aj data/multicam-dataset/0/0-Pen_B.json --cl 457

ModuleNotFoundError: No module named 'disjoint_set'

Any help for this ??
Thanks you.

dataset problem

There are two folders in the dataset link. Which one should I choose as the training set?

Custom reset function

Can you please release the code for Disjoint set custom reset function? or describe how it's supposed to work?

Can you give for me file good weights ? Thanks you

Can you share you camera information?

Hi, thanks for your sharing.
Can you share your camera information, like intrinsic parameters?
I want to map 2D images into a 3D world coordinate system.

Thanks

Can we use this for realtime cameras for person tracking?

Hi I want to implement for person tracking and reid. yolo model i need to change can you tell what i need to change for reid in your code.what i need to do to track more than two cameras

Requests for more train dataset

My research is also focused on pig-tracking. I am wondering if you could kindly send me more train dataset. I promise they will be used only for research purposed.

hi There are two folders in the link. Which one should I choose as the training set?

error with darknet

when I run tracker_cv2.py
errors appear

please help me, thanks you.

training problem

I would like to use this model for multiple-camera people-trakcing, but i see none training-guide. Is this could just use yolo-format labels to train?

Could you public your pretrained pig detection model

Thanks for you great project. I am building a system that can count pig. So can you help me, provide the pretrained weight of pig detection model.

AttributeError message when running commands from commands.txt

3840 1080
Process CameraManager-3:
Traceback (most recent call last):
File "/Library/Frameworks/Python.framework/Versions/3.10/lib/python3.10/multiprocessing/process.py", line 315, in _bootstrap
self.run()
File "/Users/multi-camera-pig-tracking/manager.py", line 79, in run
self.match_tracks_from_buffer()
File "/Users/multi-camera-pig-tracking/manager.py", line 93, in match_tracks_from_buffer
global_position_dict = self.match_tracks(self.ceiling_filter(self.local_buffer['ceiling'].pop(fid, None)),
File "/Users/multi-camera-pig-tracking/manager.py", line 152, in match_tracks
self.union_find.reset(ceil_id, min_id)
AttributeError: 'DisjointSet' object has no attribute 'reset'