Validating: 93%|██████████████████████████████████████ | 284/306 [01:52<00:08, 2.53it/s]
Training: 30%|██████████▏ | 29999/100000 [1:39:01<3:51:04, 5.05it/s]
Traceback (most recent call last):
File "/home/dev/working/train.py", line 66, in <module>
main()
File "/home/dev/working/train.py", line 62, in main
trainer.train()
File "/home/dev/working/megane2/trainer.py", line 118, in train
metrics = self.validate()
File "/usr/local/lib/python3.9/site-packages/torch/autograd/grad_mode.py", line 27, in decorate_context
return func(*args, **kwargs)
File "/home/dev/working/megane2/trainer.py", line 155, in validate
avg_f1_score.append(scores.f1_score(polygons, target_polygons))
File "/home/dev/working/megane2/scores.py", line 55, in f1_score
tp, fp, fn = get_tp_fp_fn(predicts, targets, iou_threshold)
File "/home/dev/working/megane2/scores.py", line 30, in get_tp_fp_fn
ious = [
File "/home/dev/working/megane2/scores.py", line 31, in <listcomp>
(i, j, get_polygons_iou(p1, p2))
File "/home/dev/working/megane2/scores.py", line 6, in get_polygons_iou
p1 = Polygon(p1)
File "/home/dev/.local/lib/python3.9/site-packages/shapely/geometry/polygon.py", line 255, in __new__
geom = shapely.polygons(shell, holes=holes)
File "/home/dev/.local/lib/python3.9/site-packages/shapely/decorators.py", line 77, in wrapped
return func(*args, **kwargs)
File "/home/dev/.local/lib/python3.9/site-packages/shapely/creation.py", line 251, in polygons
geometries = linearrings(geometries)
File "/home/dev/.local/lib/python3.9/site-packages/shapely/decorators.py", line 77, in wrapped
return func(*args, **kwargs)
File "/home/dev/.local/lib/python3.9/site-packages/shapely/creation.py", line 173, in linearrings
return lib.linearrings(coords, out=out, **kwargs)
shapely.errors.GEOSException: IllegalArgumentException: Points of LinearRing do not form a closed linestring
stack trace: