denoising-diffusion-gan icon indicating copy to clipboard operation
denoising-diffusion-gan copied to clipboard

How to fine tune model on my own dataset?

Open fffffq99 opened this issue 1 year ago • 0 comments

Hello. First of all, I want to thank you for your great work. I'm really interested in your work and i want to fine tune your pre-trained weights on my own dataset. as far as i know you just published the weights for NetG (Generator) model. However, in your code in order to finetune it you must have contant.pt which has weights for discriminator and some other things. so could you please share contant.pt with me?

fffffq99 avatar Jul 07 '23 12:07 fffffq99

If you want the vertices of the object mesh, you can retrieve it once you load the mesh with trimesh (e.g., here):

mesh = trimesh.load(obj_file)
print(mesh.vertices)

You can transform the vertices to another frame with apply_transform(). For example, to transform the vertices to the camera frame here, you can do:

mesh = trimesh.load(obj_file[0])
mesh.apply_transform(np.vstack((pose_y[0], [0.0, 0.0, 0.0, 1.0])))
print(mesh.vertices)

ychao-nvidia avatar Apr 07 '23 16:04 ychao-nvidia