PRNet icon indicating copy to clipboard operation
PRNet copied to clipboard

About Training

Open jhj7905 opened this issue 6 years ago • 4 comments

@YadiraF @developer-mayuan @nilday

Hello I have a question about train When i trained the network, the outputs of prediction are same as input(position map) like below temp i thought the problem was related to loss function(i mean 'weight') Parts of my code like below `uv_weight_mask = cv2.imread('/home/hyunjo/3D_Mesh/prnet_train/PRNet/Data/uv-data/uv_face_mask.png') uv_weight_mask = cv2.cvtColor(uv_weight_mask, cv2.COLOR_BGR2RGB) uv_array = np.array(uv_weight_mask, dtype=np.float32) uv_array = uv_array/255.0

x = tf.placeholder(tf.float32, shape=[None, 256, 256, 3]) label = tf.placeholder(tf.float32, shape=[None, 256, 256, 3]) weight = tf.placeholder(tf.float32, shape=[None, 256, 256, 3])

net = resfcn256(256, 256) x_op = net(x, is_training=True) loss = tf.losses.mean_squared_error(label, x_op, weights=weight)` Could you tell me advice? Thank you in advance

jhj7905 avatar Apr 01 '19 02:04 jhj7905

@jhj7905 Hello,I am very similar to your training, with the same weight. There are also some problems with the results. Did you solved this problem? My image and model do not matched.like this: image image

yexiaoxi01 avatar Apr 18 '19 11:04 yexiaoxi01

@jhj7905 @yexiaoxi01 Hi,as a newer to this, I failed to achieve the training code, could you please provide training code to me? I hope I can get your help, thank you very much. My email: [email protected]

qiannvMao avatar Jul 18 '19 04:07 qiannvMao

@qiannvMao i just try the traing code provide by https://github.com/YadiraF/PRNet/issues/46#issuecomment-455841527

yexiaoxi01 avatar Jul 19 '19 05:07 yexiaoxi01

@yexiaoxi01 thank you, I will try.

qiannvMao avatar Jul 19 '19 08:07 qiannvMao