Machine-Learning-Collection
Machine-Learning-Collection copied to clipboard
A resource for learning about Machine learning & Deep Learning
Add test function to generate the pic of val_data
Hello Sir, My name is Priyanshu Patel. I was learning GANs From your yt videos, when i was going through the code of DCGANs implementation i saw that you comment...
Hi just wanna report a small mistake in YOLO dataset.py ```python class VOCDataset(...): def __init__(...): self.annotations = pd.read_csv(csv_file) ``` should be self.annotations = pd.read_csv(csv_file, header=None), otherwise the first line of...
Hi I am using some of your codein a manuscript and I would like to give you credit. Can you update this repository with a cff https://docs.github.com/en/repositories/managing-your-repositorys-settings-and-features/customizing-your-repository/about-citation-files Thanks
Hello! Your video is very nice but I still have some trouble when training. I met "RuntimeError: shape '[64, 1024, 8, 128]' is invalid for input of size 65536" when...
[train.py](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO/train.py#L78) set the model training optimizer to Adam. ``` def main(): model = Yolov1(split_size=7, num_boxes=2, num_classes=20).to(DEVICE) optimizer = optim.Adam( model.parameters(), lr=LEARNING_RATE, weight_decay=WEIGHT_DECAY ) ``` According to the v1 paper, the...
## Code [dataset.py](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/object_detection/YOLO/dataset.py#L64-L71) calculate the```width_cell``` and ```height_cell``` to be set to the ```label_matrix``` Tensor. ``` """ ... Then to find the width relative to the cell is simply: width_pixels/cell_pixels, simplification...
I recently decided to try to make a YOLO V1 implementation as my first serious project, based on your guide, but doing all the pre-training and training the full model...
I was doing the pytorch tensor board tutorial. While running the `pytorch_tensorboard_.py` [notebook file](https://github.com/aladdinpersson/Machine-Learning-Collection/blob/master/ML/Pytorch/Basics/pytorch_tensorboard_.py), I get this: > warning: Embedding dir exists, did you set global_step for add_embedding()? [Somebody else...