denoising-diffusion-gan
denoising-diffusion-gan copied to clipboard
How to fine tune model on my own dataset?
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?
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)