Closed abhigoku10 closed 3 years ago
datasets.py
, e.g. attr_nums
, description
, and the corresponding dataset (just like what we did for RAP/PETA/PA100K). In addition, self.weights
in main.py
should also be modified according to statistics on your own dataset.attr_nums
in datasets.py
.@chufengt thanks for the response
For self.weights, just calculate the ratio of positive labels for each attribute.
@chufengt oh wokay shall do , below are additional queries 1.approximately how many number of images /data is required for training for per attribute to obtain good results
@chufengt if finetune on custom data (small number), for example after loading pretrained checkpoint using --resume config , i think the code updates all netowrk layers , not freezing any layer do you recommend that or freeze some layers and if freeze layers what layers do you think should be important and contain general info and should not be updated
@Mohamed209 Sorry I haven't tried but you can try them both.
@chufengt