When I tried to run the command
!python train.py --dataset csv --csv_train ../training.csv --csv_classes ../wheat_head_class_list.csv --csv_val ../validation.csv
got the error:
**CUDA available: True
Traceback (most recent call last):
File "/content/pytorch-retinanet/retinanet/dataloader.py", line 166, in _parse
return function(value)
ValueError: invalid literal for int() with base 10: '245.0'
During handling of the above exception, another exception occurred:
Traceback (most recent call last):
File "train.py", line 180, in
main()
File "train.py", line 57, in main
transform=transforms.Compose([Normalizer(), Augmenter(), Resizer()]))
File "/content/pytorch-retinanet/retinanet/dataloader.py", line 153, in init
self.image_data = self._read_annotations(csv.reader(file, delimiter=','), self.classes)
File "/content/pytorch-retinanet/retinanet/dataloader.py", line 270, in _read_annotations
x1 = self._parse(x1, int, 'line {}: malformed x1: {{}}'.format(line))
File "/content/pytorch-retinanet/retinanet/dataloader.py", line 168, in _parse
raise(ValueError(fmt.format(e)), None)
TypeError: exceptions must derive from BaseException**
Here's the link to my notebook