A deep learning project for binary image Segmentation with object counting. Built with PyTorch. python train.py \ --data_dir YourDataset \ --model attention_unet \ --epochs 150 \ --batch_size 4 \ --lr ...