lmd/Pose_to_SMPL_an_230402
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update code

Browse Source
This commit is contained in:
Iridoudou
2021-08-14 16:24:48 +08:00
parent f9c327d2a2
commit a5f0cf4653
4 changed files with 38 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fit/tools/main.py
View File

@ -1,5 +1,5 @@
import numpy as np
import torch import torch
import numpy as np
from tensorboardX import SummaryWriter from tensorboardX import SummaryWriter
from easydict import EasyDict as edict from easydict import EasyDict as edict
import time import time

2
fit/tools/save.py
View File

@ -6,8 +6,8 @@ import numpy as np
import pickle import pickle
sys.path.append(os.getcwd()) sys.path.append(os.getcwd())
from display_utils import display_model
from label import get_label from label import get_label
from display_utils import display_model
def create_dir_not_exist(path): def create_dir_not_exist(path):

4
fit/tools/train.py
View File

@ -95,6 +95,6 @@ def train(smpl_layer, target,
writer.add_scalar('learning_rate', float( writer.add_scalar('learning_rate', float(
optimizer.state_dict()['param_groups'][0]['lr']), epoch) optimizer.state_dict()['param_groups'][0]['lr']), epoch)
logger.info('Train ended, min_loss = {:.9f}'.format(
logger.info('Train ended, min_loss = {:.9f}'.format(float(early_stop.min_loss))) float(early_stop.min_loss)))
return res return res