face_landmark_factory icon indicating copy to clipboard operation
face_landmark_factory copied to clipboard

Custom dataset and points

Open abhigoku10 opened this issue 5 years ago • 9 comments

@songhengyang @xuansan915 Hi thanks for an wonderful code , i had few queries

  1. Can train this more number of points like 192,168... etc . What all changes i have to make it run
  2. Can i train it on custom dataset , how many ideal number o fimages should i require to get good accuracy

thansk in advance

abhigoku10 avatar Apr 15 '19 09:04 abhigoku10

  1. You can train 192 or 168 points - model with this set of tools, for more info, read the code.
  2. You can train on other dataset, generally, after training 10,000 images, you could find the difference, training above 100,000 images, you could get ideal accuracy.

songhengyang avatar Apr 15 '19 13:04 songhengyang

@songhengyang THnaks for the response , the current network is working for img size of 64 , can it be changed to other image size. How will be accuracy affected

abhigoku10 avatar Apr 16 '19 10:04 abhigoku10

Current network could work with image of other size. Image of large size will be more accurate, but the training speed will consume more time, images of small size will affect the accuracy, but the training speed will be faster.

songhengyang avatar Apr 16 '19 12:04 songhengyang

@songhengyang THanks for the response . Below are few queries

  1. I want to train on 128x128 input image , where all should i make the changes in the code , i am using squeezenet
  2. Can i train on grayscale 16bit input image which are annotated by me

abhigoku10 avatar Apr 16 '19 15:04 abhigoku10

Suggest you check and try to modify the code by yourself, or you may wait, for I am planning to release a new version to solve these problems recently.

songhengyang avatar Apr 17 '19 02:04 songhengyang

@songhengyang sure i shall fork this repo and see if can make some changes . Just wanted to knw the tentative timeline for the next release

abhigoku10 avatar Apr 17 '19 08:04 abhigoku10

We plan to release new version next month. App will train model on Ubuntu 18 and 16, not windows.

songhengyang avatar Apr 18 '19 02:04 songhengyang

@songhengyang thanks for the update and response . If you could give me some pointers on changing the image size during training it would be helpful

abhigoku10 avatar Apr 18 '19 03:04 abhigoku10

check and change parameters in script directly.

songhengyang avatar Apr 18 '19 07:04 songhengyang