search

chaneyddtt / UDA-Animal-Pose

MIT License
102 stars 12 forks source link

visualization code #18

Open R2233 opened 1 year ago

R2233 commented 1 year ago

1.Could you please put the visualization code on Github? thanks 2.if I use my own dataset, the image must be processed as mean.pth.tar train_idxs.npy vaild_idxs.npy?

chaneyddtt commented 1 year ago

Hi @R2233:

  1. Please refer to the 'cv2_visualize_keypoints' function in the visualization code .
  2. Yes, you need to have the train and test split, which are store as train_idxs.npy and vaild_idxs.npy. Regarding the mean.pth.tar, you can compute it with your own training data.