1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
|
namespace mediapipe { namespace oms {
enum CabinObjectClass { UNKNOWN = 0, PHONE = 1, WALLET = 2, LAPTOP = 3, BAG = 4, CHILD_SEAT = 10, TOY = 11, BOTTLE = 20, CUP = 21, UMBRELLA = 22, LIGHTER = 30, POWER_BANK = 31, DOG = 40, CAT = 41, MAX_CLASSES = 50 };
struct ObjectClassConfig { CabinObjectClass class_id; std::string name; float priority; float min_confidence; bool alert_enabled; };
const std::vector<ObjectClassConfig> kCabinObjectClasses = { {PHONE, "phone", 1.0f, 0.5f, true}, {WALLET, "wallet", 1.0f, 0.5f, true}, {LAPTOP, "laptop", 1.0f, 0.6f, true}, {BAG, "bag", 0.8f, 0.5f, true}, {CHILD_SEAT, "child_seat", 1.0f, 0.7f, true}, {TOY, "toy", 0.5f, 0.5f, false}, {BOTTLE, "bottle", 0.6f, 0.5f, false}, {CUP, "cup", 0.6f, 0.5f, false}, {UMBRELLA, "umbrella", 0.5f, 0.5f, false}, {LIGHTER, "lighter", 0.9f, 0.6f, true}, {POWER_BANK, "power_bank", 0.8f, 0.5f, true}, {DOG, "dog", 1.0f, 0.7f, true}, {CAT, "cat", 1.0f, 0.7f, true}, };
} }
|