search

Dv04 / Vision_Transformer

A project based on Vision Transformer for Image Regression
MIT License
14 stars 3 forks source link

Enhance Image Preprocessing in Batch Generation #1

Closed Dv04 closed 9 months ago

Dv04 commented 9 months ago

Improve the image preprocessing steps within the batch_generator function to ensure optimal performance and accuracy.

Consider exploring and implementing advanced techniques such as:

Data Augmentation: Techniques like rotation, zooming, and flipping to augment the training dataset. Normalization: Normalize image pixel values for consistent model training. Resizing: Ensure images are resized consistently to the expected input dimensions of the Vision Transformer model.

These enhancements are crucial for maintaining a consistent input format and potentially improving the model's performance.

parth-verma7 commented 9 months ago

@Dv04 I would like to work upon this issue. Can you please assign it to me.

Dv04 commented 9 months ago

Hey @parth-verma7, glad to have you with us.

Assigning you now.

parth-verma7 commented 9 months ago

@Dv04 I have created a PR. Please have a review.