2D pose estimation pipeline for tethered Drosophila.

Project description

Drosophila 2D Pose

Load the model.

from model import Drosophila2DPose
from parser import create_parser

checkpoint_path = '/home/user/Desktop/DeepFly3D/weights/sh8_deepfly.tar'
args = create_parser().parse_args('')
model = Drosophila2DPose(checkpoint_path=checkpoint_path, **args.__dict__).cuda()

Load the data.

from inference import path2inp
from dataset import Drosophila2Dataset
from torch.utils.data import DataLoader

image_path = '/home/user/Desktop/DeepFly3D/data/test/'
inp = path2inp(image_path) # extract list of images under the folder
dat = DataLoader(Drosophila2Dataset(inp), batch_size=8)

Do the inference.

from inference import inference
points2d = inference(model, dat)

Project details

Release history Release notifications | RSS feed

0.14

Jun 21, 2022

This version

0.13

Mar 17, 2022

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

nely-df2d-0.13.tar.gz (7.3 kB view details)

Uploaded Mar 17, 2022 Source

Built Distribution

nely_df2d-0.13-py3-none-any.whl (9.1 kB view details)

Uploaded Mar 17, 2022 Python 3

File details

Details for the file nely-df2d-0.13.tar.gz.

File metadata

Download URL: nely-df2d-0.13.tar.gz
Upload date: Mar 17, 2022
Size: 7.3 kB
Tags: Source
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.8.0 pkginfo/1.8.2 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/0.9.1 urllib3/1.26.8 tqdm/4.63.0 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.4 CPython/3.6.13

File hashes

Hashes for nely-df2d-0.13.tar.gz
Algorithm	Hash digest
SHA256	`eca89f8ed80855efce002488356b92f65330b76f5e859598c7dfe55666b238b9`
MD5	`8e1c7a02b8a9a14bd328932b49a234f2`
BLAKE2b-256	`85ac64e62ec1de5c2951e48092e144d6a238d05cfef3bfb79566999c0b2115af`

See more details on using hashes here.

File details

Details for the file nely_df2d-0.13-py3-none-any.whl.

File metadata

Download URL: nely_df2d-0.13-py3-none-any.whl
Upload date: Mar 17, 2022
Size: 9.1 kB
Tags: Python 3
Uploaded using Trusted Publishing? No
Uploaded via: twine/3.8.0 pkginfo/1.8.2 readme-renderer/34.0 requests/2.27.1 requests-toolbelt/0.9.1 urllib3/1.26.8 tqdm/4.63.0 importlib-metadata/4.8.3 keyring/23.4.1 rfc3986/1.5.0 colorama/0.4.4 CPython/3.6.13

File hashes

Hashes for nely_df2d-0.13-py3-none-any.whl
Algorithm	Hash digest
SHA256	`33b4f897a92ca3cc056a8cd0ee0a13defa6dfffd20b1cbece3a4fb44c13575bd`
MD5	`60fb5776b89007389f04ee5a83292428`
BLAKE2b-256	`c6648558baf224e1d6552920fb2afdc1caa2fdc49b1076ac408e73de7db5ecb4`