supernotman / RetinaFace_Pytorch

Reimplement RetinaFace with Pytorch
305 stars 68 forks source link

About data argumentation #14

Closed ElvishElvis closed 5 years ago

ElvishElvis commented 5 years ago

Hello Which data argumentation did you use in your actual trainning? Cuz I saw several methods that you had commented but not sure which ones did you actually use.

BTW, many of them are not working and have bugs.

for example,

add this to line 297 in dataloader.py

pad = torch.from_numpy(np.array(pad)) before this padded_img = F.pad(img, pad, "constant", value=0).

Or it will show

TypeError: narrow(): argument 'start' (position 2) must be int, not numpy.int64