marando / pycatfd Goto Github PK

View Code? Open in Web Editor NEW

145.0 6.0 37.0 12.7 MB

Cat facial detection and landmark recognition in Python

Python 96.24% Dockerfile 3.12% Shell 0.64%

machine-learning neural-network deep-learning facial-recognition cats

pycatfd's Introduction

pycatfd 🐈

Cat facial detection and landmark recognition with Python

Dependencies

dlib
opencv

Python Requirements

Pillow
requests
scikit-image

Usage

Use catfd.py to detect cat faces and facial landmarks in individual images or an entire folder. The repository comes pre-trained, but can be re-trained using the train.py tool.

Docker Container

A Dockerfile is provided to avoid having to install all dependencies manually, which can be quite tedious. First make sure that you have Docker installed on your system, and then to use the image, build it using the build.sh script, and then run it using the run.sh script from within the repository directory.

If you would like to do this manually, you can run the following:

Clone and enter the repository:

git clone [email protected]:marando/pycatfd.git
cd pycatfd

Build the image:
```
docker build -t pycatfd .
```

Then issue this command to run it:

docker run -it --rm -v "$PWD":/app pycatfd /bin/bash

pycatfd's People

Contributors

Stargazers

Watchers

pycatfd's Issues

more labeled dataset?

Hi~
there're only 40 images with label
Do you have more training data? Or is there any open dataset?
Thanks!~

License

Can you please add licensing details to the project?

Is it available for commercial usage?

Custom dataset format

Hi, I'm interested in your pycatfd.
But i wanna apply it to other animals like cow, dog, pig etc.

How could make dataset format?
According to your 'read me', it looks like bataset is based on json format.

Can I make my custom dataset based on COCO?

After successful Docker build, dlib not quite working right.

First off compiling dlib (which was a pain the the arse and managed to OOM my system 5 times before I got it to build.) Anyways after a successful build, after running the docker command to enter the container, running catfd.py can't find dlib.

./catfd.py
Traceback (most recent call last):
File "./catfd.py", line 9, in
from lib.Detector import Detector
File "/app/lib/Detector.py", line 1, in
import dlib
ModuleNotFoundError: No module named 'dlib'

Log from the build

98%] Building CXX object CMakeFiles/_dlib_pybind11.dir/src/image_dataset_metadata.cpp.o
[ 99%] Building CXX object CMakeFiles/_dlib_pybind11.dir/src/numpy_returns.cpp.o
[100%] Building CXX object CMakeFiles/_dlib_pybind11.dir/src/line.cpp.o
[100%] Linking CXX shared module /root/dlib/build/lib.linux-x86_64-3.8/_dlib_pybind11.cpython-38-x86_64-linux-gnu.so
[100%] Built target _dlib_pybind11
creating build/bdist.linux-x86_64
creating build/bdist.linux-x86_64/egg
creating build/bdist.linux-x86_64/egg/dlib
copying build/lib.linux-x86_64-3.8/dlib/init.py -> build/bdist.linux-x86_64/egg/dlib
copying build/lib.linux-x86_64-3.8/_dlib_pybind11.cpython-38-x86_64-linux-gnu.so -> build/bdist.linux-x86_64/egg
byte-compiling build/bdist.linux-x86_64/egg/dlib/init.py to init.cpython-38.pyc
creating stub loader for _dlib_pybind11.cpython-38-x86_64-linux-gnu.so
byte-compiling build/bdist.linux-x86_64/egg/_dlib_pybind11.py to _dlib_pybind11.cpython-38.pyc
creating build/bdist.linux-x86_64/egg/EGG-INFO
copying tools/python/dlib.egg-info/PKG-INFO -> build/bdist.linux-x86_64/egg/EGG-INFO
copying tools/python/dlib.egg-info/SOURCES.txt -> build/bdist.linux-x86_64/egg/EGG-INFO
copying tools/python/dlib.egg-info/dependency_links.txt -> build/bdist.linux-x86_64/egg/EGG-INFO
copying tools/python/dlib.egg-info/not-zip-safe -> build/bdist.linux-x86_64/egg/EGG-INFO
copying tools/python/dlib.egg-info/top_level.txt -> build/bdist.linux-x86_64/egg/EGG-INFO
writing build/bdist.linux-x86_64/egg/EGG-INFO/native_libs.txt
creating dist
creating 'dist/dlib-19.21.99-py3.8-linux-x86_64.egg' and adding 'build/bdist.linux-x86_64/egg' to it
removing 'build/bdist.linux-x86_64/egg' (and everything under it)
Processing dlib-19.21.99-py3.8-linux-x86_64.egg
creating /usr/local/lib/python3.8/dist-packages/dlib-19.21.99-py3.8-linux-x86_64.egg
Extracting dlib-19.21.99-py3.8-linux-x86_64.egg to /usr/local/lib/python3.8/dist-packages
Adding dlib 19.21.99 to easy-install.pth file
Installed /usr/local/lib/python3.8/dist-packages/dlib-19.21.99-py3.8-linux-x86_64.egg

When I check the container, the dlib package has not made it into the container.
(/usr/local/lib/python3.8/dist-packages/dlib-19.21.99-py3.8-linux-x86_64.egg)

I don't quite yet know how to make my own docker containers, and wondering if there is something I am missing here?

Can u share a video on youtube?

Sorry about this question. I want to know if your program perform good enough.
I find it is hard to use. First, you need install opencv2( I install opencv3 before). And I need build dlib on my computer. But the most trick part is I want to move it to raspberry pi zero.

Sorry about it. If it just a samll test project. Youcan tell me. Thanks.

draw_line function parameter name change

Using OpenCV version 3.4.1
I needed to change line 221 from "cv2.CV_AA" -> "cv2.LINE_AA"

    cv2.line(img, pt1, pt2, color, width, cv2.CV_AA)

...becomes:

    cv2.line(img, pt1, pt2, color, width, cv2.LINE_AA)