litert-community/RTMPose-Animal-AP10K-LiteRT

RTMPose-Animal (AP-10K) — LiteRT (on-device animal pose, fully-GPU)

RTMPose (mmpose) animal pose, trained on AP-10K, converted to LiteRT and running fully on the CompiledModel GPU (ML Drift) on Android. Top-down: a square animal crop → 17 AP-10K keypoints (eyes, nose, neck, tail root, four limbs).

On-device (Pixel 8a, Tensor G3 — verified)

nodes on GPU	333 / 333 LITERT_CL (full residency)
inference	~5 ms (256×256)
size	27.5 MB (fp16)
accuracy	device-vs-PyTorch SimCC corr 0.999, 17/17 keypoints

image[1,3,256,256] (mmpose mean/std) →[GPU: RTMPose-m]→ simcc_x[1,17,512], simcc_y[1,17,512]

Output[0] = simcc_x, output[1] = simcc_y; each keypoint = argmax over its 1D SimCC (bins = pixels × 2).

Minimal usage

Android (Kotlin, CompiledModel GPU)

val model = CompiledModel.create(context.assets, "rtm_animal_fp16.tflite",
    CompiledModel.Options(Accelerator.GPU), null)
val inputs = model.createInputBuffers()
val outputs = model.createOutputBuffers()
inputs[0].writeFloat(chw)              // [1,3,256,256] mmpose mean/std (0-255 RGB), NCHW
model.run(inputs, outputs)
val simccX = outputs[0].readFloat()    // [1,17,512]
val simccY = outputs[1].readFloat()    // [1,17,512]; keypoint = argmax / 2

Python (desktop verification)

MEAN = np.array([123.675, 116.28, 103.53], np.float32)
STD  = np.array([58.395, 57.12, 57.375], np.float32)
import numpy as np
from PIL import Image
from ai_edge_litert.interpreter import Interpreter

img = Image.open("dog.jpg").convert("RGB").resize((256, 256))  # centered subject crop
x = ((np.asarray(img, np.float32) - MEAN) / STD).transpose(2, 0, 1)[None]

it = Interpreter(model_path="rtm_animal_fp16.tflite"); it.allocate_tensors()
it.set_tensor(it.get_input_details()[0]["index"], x); it.invoke()
od = it.get_output_details()                                     # output 0 = simcc_x, 1 = simcc_y
sx = it.get_tensor(od[0]["index"])[0]                             # simcc_x [17,512]
sy = it.get_tensor(od[1]["index"])[0]                             # simcc_y [17,512]
kx, ky = sx.argmax(-1) / 2.0, sy.argmax(-1) / 2.0                 # 17 keypoints, px in 256x256
for i, (a, b) in enumerate(zip(kx, ky)):
    print(f"kp{i}: ({a:.1f}, {b:.1f})")

How it converts (litert-torch) — the RTMPose recipe, unchanged

Same model family as the human-pose RTMPose; only the config/checkpoint change to AP-10K. The two on-device-only Mali fixes transfer without modification: ScaleNorm → SafeRMSNorm (the RMS-norm channel Σx² overflows fp16 on Mali → scale down by 64 before squaring, then rescale) and GAU act@act BMM → broadcast-reduce. Result: banned ops NONE, ≤4D, tflite-vs-torch corr 1.0, device-vs-torch 0.999.

Preprocessing

Center-crop to a square, resize 256×256, mmpose mean/std (RGB, 0-255 scale), NCHW.

License

Apache-2.0. Upstream: open-mmlab/mmpose; dataset AP-10K.