levindabhi/cloth-segmentation
This repo contains code and a pre-trained model for clothes segmentation.
PythonMIT
Issues
- 1
- 14
- 1
- 14
Inference error on trained checkpoints
#17 opened by richard-schwab - 6
Getting the predictions
#8 opened by Lord-Psarris - 2
Prorosal: Pack model for Huggingface inference
#15 opened by betafuzz - 2
Is it possible to segregate the final output into each seperate label and have its original colour from the input image?
#16 opened by mdrameez1412 - 3
AttributeError: 'NoneType' object has no attribute 'to'; Issue with colab
#14 opened by AlexanderKozhevin - 0
final_label = first_channel + second_channel * 2 + third_channel * 3 conflict_mask = (final_label <= 3).astype("uint8") final_label = (conflict_mask) * final_label + (1 - conflict_mask) * 1 target_tensor = torch.as_tensor(final_label, dtype=torch.int64)
#18 opened by DWCTOD - 0
[question] Segmentation mask dimensions
#13 opened by b4nn3d - 0
[bug] labels should start from 1
#12 opened by dmascali - 0
Possible to adjust sensitivity threshold?
#11 opened by ideavr - 1
RuntimeError: CUDA out of memory.
#10 opened by mayanktiwariiiitdmj - 0
Can pretrain model to 11 class ?
#9 opened by ThanhJamie - 15
- 4
TypeError: Caught TypeError in DataLoader worker process 0. and TypeError: 'float' object cannot be interpreted as an integer
#6 opened by davichen2017 - 1
IsADirectoryError: [Errno 21] Is a directory: 'input_images/.ipynb_checkpoints'
#5 opened by basemdabbour - 1
Model isn't accurate
#3 opened by AndreyStille - 2
Model very slow on cpu
#1 opened by AndreyStille - 0
您好!如何将分割结果输出为RGB彩色图?
#2 opened by Eddierr