from deeplabcut.utils import auxiliaryfunctions
cfg = auxiliaryfunctions.read_config(config) #
modelfoldername = auxiliaryfunctions.get_model_folder(
cfg["TrainingFraction"][trainingsetindex], shuffle, cfg, modelprefix=modelprefix
)
trainposeconfigfile = Path(
os.path.join(
cfg["project_path"], str(modelfoldername), "train", "pose_cfg.yaml"
)
)
testposeconfigfile = Path(
os.path.join(cfg["project_path"], str(modelfoldername), "test", "pose_cfg.yaml")
)
snapshotfolder = Path(
os.path.join(cfg["project_path"], str(modelfoldername), "train")
)
return trainposeconfigfile, testposeconfigfile, snapshotfolder
if allow_growth:
os.environ["TF_FORCE_GPU_ALLOW_GROWTH"] = "true"
import tensorflow as tf
# reload logger.
import importlib
import logging
importlib.reload(logging)
logging.shutdown()
from deeplabcut.utils import auxiliaryfunctions
tf.compat.v1.reset_default_graph()
start_path = os.getcwd()
# Read file path for pose_config file. >> pass it on
cfg = auxiliaryfunctions.read_config(config) #2
modelfoldername = auxiliaryfunctions.get_model_folder(
cfg["TrainingFraction"][trainingsetindex], shuffle, cfg, modelprefix=modelprefix
)
poseconfigfile = Path(
os.path.join(
cfg["project_path"], str(modelfoldername), "train", "pose_cfg.yaml"
)
)
curl http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz | tar xvz