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-07 21:32:53 +08:00
parent e68d723056
commit 1a5646d843
3 changed files with 1 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fit/configs/UTD_MHAD.json
View File

@ -4,7 +4,7 @@
}, },
"TRAIN": { "TRAIN": {
"LEARNING_RATE": 2e-2, "LEARNING_RATE": 2e-2,
"MAX_EPOCH": 10, "MAX_EPOCH": 500,
"WRITE": 1 "WRITE": 1
}, },
"USE_GPU": 1, "USE_GPU": 1,

3
fit/tools/main.py
View File

@ -1,16 +1,13 @@
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
import inspect
import sys import sys
import os import os
import logging import logging
import argparse import argparse
import json import json
from tqdm import tqdm
sys.path.append(os.getcwd()) sys.path.append(os.getcwd())
from display_utils import display_model from display_utils import display_model
from smplpytorch.pytorch.smpl_layer import SMPL_Layer from smplpytorch.pytorch.smpl_layer import SMPL_Layer

21
fit/tools/train.py
View File

@ -1,32 +1,11 @@
from fit.tools.save import save_pic
import matplotlib as plt
from matplotlib.pyplot import show
import torch import torch
import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
from torch.nn.modules import module
from torch.optim import lr_scheduler
import torch.optim as optim import torch.optim as optim
from torch.utils.data import DataLoader
import torchvision.datasets as dset
import torchvision.transforms as T
import numpy as np
import scipy.io
from tensorboardX import SummaryWriter
from easydict import EasyDict as edict
import time
import inspect
import sys import sys
import os import os
import logging
import argparse
import json
from tqdm import tqdm from tqdm import tqdm
sys.path.append(os.getcwd()) sys.path.append(os.getcwd())
from smplpytorch.pytorch.smpl_layer import SMPL_Layer
from display_utils import display_model
from map import mapping
class Early_Stop: class Early_Stop: