Machine-Learning-Collection VS 6DRepNet

def compute_mean_std(dataloader): ''' We assume that the images of the dataloader have the same height and width source: https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/Basics/pytorch_std_mean.py ''' # var[X] = E[X**2] - E[X]**2 channels_sum, channels_sqrd_sum, num_batches = 0, 0, 0 for batch_images, labels in tqdm(dataloader): # (B,H,W,C) batch_images = batch_images.permute(0,3,4,2,1) channels_sum += torch.mean(batch_images, dim=[0, 1, 2, 3]) channels_sqrd_sum += torch.mean(batch_images ** 2, dim=[0, 1, 2,3]) num_batches += 1 mean = channels_sum / num_batches std = (channels_sqrd_sum / num_batches - mean ** 2) ** 0.5 return mean, std compute_mean_std(dataloader)

I then found Aladdin Persson implementation (which he described in YouTube video). He said that original paper used Dropout, because BatchNorm was not invented at the time, and he wants to use BatchNorm instead. I thought there is no critical difference between these two, and decided to stick up with paper for the sake of learning to implement such things.

I am trying to train a model for NMT on a custom dataset. I found this great tutorial on youtube along with the accompanying repo, but it uses an old version of PyTorch and torchtext. More recent versions of torchtext have removed the Field and BucketIterator classes. I looked for more recent tutorials. The closest thing I could find was this medium post (again with the accompanying code) which worked with a custom dataset for text classification. I tried to replicate the code with my problem and got this far:

with the source code here: Machine-Learning-Collection/ML/Pytorch/GANs/CycleGAN at master · aladdinpersson/Machine-Learning-Collection · GitHub

seq2seq_attention

I followed a tutorial to do a pix2pix GAN network here: https://www.youtube.com/watch?v=SuddDSqGRzg, and the github.

Aladdin Persson

Hello everyone. I am all set up and ready to go. I downloaded the code below (btw, I don't fully understand what it does, lol) and ran it a few times.

Code: https://github.com/aladdinpersson/Machine-Learning-Collection/tree/master/ML/Pytorch/object_detection/YOLOv3

Well so it sounds like you want to do something like DeepFaceLab? I don't have a lot of experience with that but I believe they collect facial images based on their angles (pitch, roll and yaw) and match and merge them together. So something like this might help you there: https://github.com/thohemp/6DRepNet

I found this was the easiest one to get installed and up and running: https://github.com/thohemp/6DRepNet

Code for https://arxiv.org/abs/2202.12555 found: https://github.com/thohemp/6DRepNet