All information regarding the third homework assignment can be found here: UROB HW3 – Fruit Image Analysis Repository

This homework focuses on training a Convolutional Neural Network (CNN) to:

• 🏷️ Classify images of fruits into 30 categories
• 🎨 Segment fruits using pixel-level masks
• 🔍 Learn meaningful image embeddings

The goal is to design, train, and evaluate a model capable of performing all three tasks efficiently.

• Implement your CNN model in `model.py` as class `MyModel`
• The `forward()` method must return:
  1. Class logits `[batch_size, 30]`
  2. Segmentation mask `[batch_size, 1, 64, 64]`
  3. Image embeddings `[batch_size, embedding_dim]`
• Implement `get_embedding()` returning embeddings only
• Complete the training loop in `train.py`
• Tune hyperparameters in `confs/config.yml`

Basic Evaluation (10 points)

Tournament Evaluation (up to +3 bonus points) Models are ranked by performance across all tasks. Lower total rank = better score. Ties are broken by earlier submission.

* 🥇 1st-3rd place → +3 pts

* 🥈 4th-6th place → +2 pts

* 🥉 7th-9th place → +1 pts

Maximum total = 13 points

Submit a ZIP file containing:

• `model.py` – your model
• `train.py` – your training script
• `weights.pth` – trained model weights

The size limit for a .zip file is 500 MB.

Pretrained Models Policy ❌ No pretrained models or transfer learning allowed. ✅ Train from scratch using only the provided dataset.

Plagiarism Policy ✅ You may discuss ideas with classmates. ❌ Direct code copying = 0 points.

Keep the data loader similar to lab06. The images for the evaluation will be loaded and preprocessed in the same way.

Feel free to add data augmentation for better results.

Do not create giant networks for two reasons:

*It is not needed for this task :)

*The upload system allows you to upload a .zip file up to 500 MB. If your weights are larger than 500 MB, they cannot be evaluated.

Don't be shy using the compute. The cluster is for you; use it as much as possible to test your ideas.

Classification

AlexNet VGGNet ResNet

Segmentation

FCN U-Net Mask-RCNN

Good luck! 🍀 For questions, contact: jan.skvrna@cvut.cz