96da2cacf837a9b84ecdeafb50dfb4a93b402f33,deeplabcut/pose_estimation_tensorflow/nnet/net_factory.py,,pose_net,#Any#,12

Before Change

    net_type = cfg.net_type
    if "mobilenet" in net_type:  // multi currently not supported
        print("Initializing MobileNet")
        if cfg.dataset_type == "multi-animal-imgaug":
            raise Exception(
                "MobileNets are currently not yet supported for multianimal DLC!"
            )

        from deeplabcut.pose_estimation_tensorflow.nnet.pose_net_mobilenet import (
            PoseNet,
        )

After Change

            cls = PoseNet
    elif "efficientnet" in net_type:
        if (
            cfg.get("stride", 8) < 8
        ):  // this supports multianimal (with PAFs) or pairwise prediction
            from deeplabcut.pose_estimation_tensorflow.nnet.pose_netmulti import PoseNet

            cls = PoseNet
        else:
            print("Initializing Efficientnet")
            from deeplabcut.pose_estimation_tensorflow.nnet.pose_net_efficientnet import PoseNet
            cls = PoseNet
    else:
        raise Exception("Unsupported class of network: "{}"".format(net_type))

    return cls(cfg)

Instances